Update app.py

app.py

@@ -53,14 +53,12 @@ def predict(img):
     as a dictionary:
         {label: confidence, label: confidence, ...}
     """
-    print("Input type:", type(img))
-    print("Input type:", img.keys())
 
     try:
         if img is None or not isinstance(img, dict) or 'image' not in img:
             return {"Error": 1.0}
     
-        img_data = img['layers'][0]
+        img_data = img['composite']
         img_gray = Image.fromarray(img_data).convert('L').resize((28, 28))
         img_tensor = transforms.ToTensor()(img_gray).unsqueeze(0)
     
@@ -78,7 +76,12 @@ def predict(img):
 
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.Sketchpad(label="Draw Here", brush=gr.Brush(default_size=20, default_color="#000000", colors=["#000000"]), image_mode="L"),
+    inputs=gr.Sketchpad(
+        label="Draw Here",
+        brush=gr.Brush(default_size=20, default_color="#FFFFFF", colors=["#FFFFFF"]),
+        image_mode="L",
+        layers=False
+    ),
     outputs=gr.Label(label="Guess"),
     title="Thai Digit Handwritten Classification",
     description="ทดลองวาดภาพตัวอักษรเลขไทยลงใน Sketchpad ด้านล่างเพื่อทำนายผลตัวเลข ตั้งแต่ ๐ (ศูนย์) ๑ (หนึ่ง) ๒ (สอง) ๓ (สาม) ๔ (สี่) ๕ (ห้า) ๖ (หก) ๗ (เจ็ด) ๘ (แปด) จนถึง ๙ (เก้า)",