Update app.py

app.py

@@ -15,7 +15,7 @@ if not os.path.exists("results"):
 def infer(img):
     #img = ImageOps.contain(img, (700, 700))
     width, height = img.size
-    res=np.ones_like(img.shape)
+    res=np.ones_like(img.size)
     print(img.shape)
     print(width)
     img.save("./data/data.png")