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", 1.0)]
+            return {"Error": 1.0}
     
         img_data = img['image']
         img_gray = Image.fromarray(img_data).convert('L').resize((28, 28))
@@ -69,15 +69,15 @@ def predict(img):
     
         probs, indices = torch.topk(probs, 5)  # select top 5
         probs, indices = probs.tolist(), indices.tolist()  # transform to list
-        return [(LABELS[i], float(v)) for i, v in zip(indices, probs)]
+        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", 1.0)]
+        return {"Error": 1.0}
 
 
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.Sketchpad(label="Draw Here", brush=gr.Brush(default_size=20, default_color="#000000", colors=["#000000"])),
+    inputs=gr.Sketchpad(label="Draw Here", brush=gr.Brush(default_size=20, default_color="#000000", colors=["#000000"]), image_mode="L"),
     outputs=gr.Label(label="Guess"),
     title="Thai Digit Handwritten Classification",
     description="ทดลองวาดภาพตัวอักษรเลขไทยลงใน Sketchpad ด้านล่างเพื่อทำนายผลตัวเลข ตั้งแต่ ๐ (ศูนย์) ๑ (หนึ่ง) ๒ (สอง) ๓ (สาม) ๔ (สี่) ๕ (ห้า) ๖ (หก) ๗ (เจ็ด) ๘ (แปด) จนถึง ๙ (เก้า)",