Update app.py

app.py

@@ -53,9 +53,7 @@ def predict(img):
     as a dictionary:
         {label: confidence, label: confidence, ...}
     """
-
     img_data = img['composite']
-    print(img_data.sum())
     img_gray = Image.fromarray(img_data).convert('L').resize((28, 28))
     img_tensor = transforms.ToTensor()(img_gray).unsqueeze(0)
 
@@ -72,9 +70,10 @@ demo = gr.Interface(
     fn=predict,
     inputs=gr.Sketchpad(
         label="Draw Here",
-        brush=gr.Brush(default_size=20, default_color="#000000", colors=["#000000"]),
+        brush=gr.Brush(default_size=20, default_color="#FFFFFF", colors=["#FFFFFF"]),
         image_mode="L",
-        layers=False
+        layers=False,
+        eraser=None,
     ),
     outputs=gr.Label(label="Guess"),
     title="Thai Digit Handwritten Classification",