Update app.py

app.py

@@ -55,7 +55,7 @@ def predict(img):
     image = transforms.Resize((480,640))(img)
     tensor = transform_image(image=image)
     with torch.no_grad():
-        outputs = model(tensor)
+        outputs = learn.model(tensor)
     
     outputs = torch.argmax(outputs,1)