update

app.py

@@ -31,8 +31,8 @@ def transform_image(img_sample):
 def predict(Image):
     tranformed_img = transform_image(Image)
     model.eval()
-    tranformed_img = transform_image(Image)
-    img = torch.from_numpy(tranformed_img)
+    img = transform_image(Image)
+    #img = torch.from_numpy(tranformed_img)
 
     with torch.no_grad():
         grade = torch.softmax(model(img.float()), dim=1)[0]