Update app.py

app.py

@@ -57,7 +57,7 @@ def predict(img):
 
     try:
         if img is None or not isinstance(img, dict) or 'image' not in img:
-            return {"Error": "Invalid input"}
+            return [("Error", 1.0)]
     
         img_data = img['image']
         img_gray = Image.fromarray(img_data).convert('L').resize((28, 28))
@@ -69,11 +69,10 @@ def predict(img):
     
         probs, indices = torch.topk(probs, 5)  # select top 5
         probs, indices = probs.tolist(), indices.tolist()  # transform to list
-        confidences = {LABELS[i]: float(v) for i, v in zip(indices, probs)}
-        return confidences
+        return [(LABELS[i], float(v)) for i, v in zip(indices, probs)]
     except Exception as e:
         print(f"Error in prediction: {str(e)}")
-        return {"Error": str(e)}
+        return [("Error", 1.0)]
 
 
 demo = gr.Interface(