Push to hf

handler.py

@@ -279,7 +279,7 @@ class EndpointHandler:
 
         print(self.inference_progress)
 
-    def check_progress(self, request_id: str) -> dict[str, str | float]:
+    def check_progress(self, request_id: str) -> Dict[str, Union[str, float]]:
         progress = self.inference_progress.get(request_id, 0)
         latest_image = self.inference_images.get(request_id, None)