Update app.py

app.py

@@ -12,6 +12,7 @@ from pathlib import Path
 import random
 import PIL
 import torchvision.transforms as transformss
+import torch
 from albumentations import *
 
 def get_y_fn (x):
@@ -76,7 +77,7 @@ def transform_image(image):
 def predict(img):
     img = PIL.Image.fromarray(img, "RGB")
     image = transformss.Resize((480,640))(img)
-    tensor = transform_image(image=img) 
+    tensor = transform_image(image=image) 
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     learner.to(device)