Update app.py

app.py

@@ -39,6 +39,7 @@ def recognize_digit(image):
                         transforms.ToTensor(),
                         transforms.Normalize((0.5,), (0.5,))
                     ])
+        image = PIL.ImageOps.invert(image)  # Invert colors
         image = transform(image)
         with torch.inference_mode(): # inference mode of pytoroch
             prediction = load_model(image)
@@ -49,6 +50,6 @@ def recognize_digit(image):
     
 
 demo = gr.Interface(fn=recognize_digit, 
-                    inputs=gr.Image(image_mode="L", invert_colors=True, source="canvas"), 
+                    inputs=gr.Image(image_mode="L", source="canvas"), 
                     outputs=gr.Label(num_top_classes=1))
 demo.launch(share=True)