Update app.py

app.py

@@ -33,24 +33,23 @@ model = CNN()
 model.load_state_dict(torch.load("model.pth",map_location="cpu"))
 model.eval()
 
-def predict(image):
+def predict(img):
 
-    image = np.array(image)
-    image = image/255.0
-    image = image.reshape(1,1,28,28)
+    img = img / 255.0
+    img = np.resize(img, (28,28))
+    img = img.reshape(1,1,28,28)
 
-    image = torch.tensor(image,dtype=torch.float32)
+    img = torch.tensor(img, dtype=torch.float32)
 
-    output = model(image)
+    output = model(img)
     pred = torch.argmax(output,1).item()
 
-    return pred
+    return str(pred)
 
 interface = gr.Interface(
     fn=predict,
-    inputs=gr.Image(shape=(28,28),image_mode="L"),
+    inputs=gr.Image(type="numpy", image_mode="L"),
     outputs="label",
     title="Handwritten Digit Classifier"
 )
-
 interface.launch()