Update app.py

app.py

@@ -1077,18 +1077,33 @@ async def submit(
                 return None
         progress_stub = _NoProgress()
 
-        output_files, zip_files = _api_upscale.inference(
-            gallery=gallery,
-            face_restoration=face_model if face_model != "None" else None,
-            upscale_model=upscale_model if upscale_model != "None" else None,
-            scale=float(scale),
-            face_detection=face_detection,
-            face_detection_threshold=face_detection_threshold,
-            face_detection_only_center=face_detection_only_center,
-            outputWithModelName=with_model_name,
-            save_as_png=True,
-            progress=progress_stub,
-        )
+        # output_files, zip_files = _api_upscale.inference(
+        #     gallery=gallery,
+        #     face_restoration=face_model if face_model != "None" else None,
+        #     upscale_model=upscale_model if upscale_model != "None" else None,
+        #     scale=float(scale),
+        #     face_detection=face_detection,
+        #     face_detection_threshold=face_detection_threshold,
+        #     face_detection_only_center=face_detection_only_center,
+        #     outputWithModelName=with_model_name,
+        #     save_as_png=True,
+        #     progress=progress_stub,
+        # )
+        try:
+            output_files, zip_files = _api_upscale.inference(
+                gallery=gallery,
+                face_restoration=face_model if face_model != "None" else None,
+                upscale_model=upscale_model if upscale_model != "None" else None,
+                scale=float(scale),
+                face_detection=face_detection,
+                face_detection_threshold=face_detection_threshold,
+                face_detection_only_center=face_detection_only_center,
+                outputWithModelName=with_model_name,
+                save_as_png=True,
+                progress=progress_stub,
+            )
+        except Exception as infer_err:
+            raise RuntimeError(f"Inference failed: {infer_err}")  # FORCE FAIL
         # -----------------------------------------------------------------
         # NEW: MEDIA CLICK LOGGING (AFTER SUCCESSFUL CORE PROCESSING)
         # -----------------------------------------------------------------