Update app.py

app.py

@@ -32,7 +32,6 @@ transform = transforms.Compose([
 ])
 
 
-# ── Inference logic ──
 def classify(image, progress=gr.Progress(track_tqdm=True)):
     try:
         image = image.convert("RGB")
@@ -44,7 +43,10 @@ def classify(image, progress=gr.Progress(track_tqdm=True)):
 
         return {CLASS_NAMES[prediction]: 1.0}
     except Exception as e:
-        return {"Error": str(e)}
+        print("Error during prediction:", e)
+        # Return fixed label with zero confidence or default fallback
+        return {"Error": 0.0}
+
 
 
 # ── Gradio Layout ──