Update app.py

app.py

@@ -53,6 +53,10 @@ def predict(img):
     as a dictionary:
         {label: confidence, label: confidence, ...}
     """
+    if img.get("composite") is not None:
+        if img["composite"].sum() == 0:
+            return {"No input sketch": 0.0}
+
     img_data = img['composite']
     img_gray = Image.fromarray(img_data).convert('L').resize((28, 28))
     img_tensor = transforms.ToTensor()(img_gray).unsqueeze(0)
@@ -85,6 +89,8 @@ with gr.Blocks(js=js_func) as demo:
             image_mode="L",
             layers=False,
             eraser=None,
+            width=400,
+            height=400
         ),
         outputs=gr.Label(label="Guess"),
         title="Thai Digit Handwritten Classification",