update

app.py

@@ -32,7 +32,11 @@ def predict(Image):
     tranformed_img = transform_image(Image)
     model.eval()
     img = transform_image(Image)
+    img = img.reshape(1,3,224,224)
     #img = torch.from_numpy(tranformed_img)
+    #outputs = model(img)
+    #class_out = outputs.argmax(dim=1)
+
 
     with torch.no_grad():
         grade = torch.softmax(model(img.float()), dim=1)[0]