Update app.py

app.py

@@ -60,7 +60,7 @@ def predict(img):
         if img is None or not isinstance(img, dict) or 'image' not in img:
             return {"Error": 1.0}
     
-        img_data = img['image']
+        img_data = img['layers'][0]
         img_gray = Image.fromarray(img_data).convert('L').resize((28, 28))
         img_tensor = transforms.ToTensor()(img_gray).unsqueeze(0)