Update app.py

app.py

@@ -189,7 +189,8 @@ model_manager = ModelManager(MODEL_CATALOG)
 
 # --- Prediction Function (v3 from before) ---
 def predict_anatomy_v3(image: Image.Image, model_name: str):
-    if image is None: return {"Error": "No image provided"}
+    if image is None:
+        return {"Error": 1.0, "Info": 0.0}  # Return numeric values
     try:
         model_manager.load_model(model_name)
         pil_image = image.convert("RGB")
@@ -245,7 +246,12 @@ def predict_anatomy_v3(image: Image.Image, model_name: str):
         return output_probs
     except Exception as e:
         print(f"Error during prediction: {e}\n{traceback.format_exc()}")
-        return {"Error": str(e)}
+        # Return properly formatted error for Gradio Label
+        error_msg = str(e)[:50] + "..." if len(str(e)) > 50 else str(e)
+        return {
+            f"Error: {error_msg}": 1.0,
+            "Please check logs": 0.0
+        }
 
 # --- Gradio Interface ---
 DESCRIPTION = """