Upload handler.py

handler.py

@@ -1,5 +1,5 @@
 import os
-from typing import Any, Dict, Tuple
+from typing import Any, Dict, Union
 from PIL import Image
 import torch
 from diffusers import FluxPipeline
@@ -48,7 +48,7 @@ class EndpointHandler:
         time_taken = end_time - start_time
         print(f"Time taken: {time_taken:.2f} seconds")
 
-    def __call__(self, data: Dict[str, Any]) -> Tuple[Image.Image, None]:
+    def __call__(self, data: Dict[str, Any]) -> Union[Image.Image, None]:
         try:
             logger.info(f"Received incoming request with {data=}")