Update app.py

app.py

@@ -56,7 +56,8 @@ def predict(img):
     if img is None:
         return None
 
-    img_gray = Image.fromarray(img).convert('L').resize((28, 28))
+    img_data = img['image']
+    img_gray = Image.fromarray(img_data).convert('L').resize((28, 28))
     img_tensor = transforms.ToTensor()(img_gray).unsqueeze(0)
 
     # Make prediction