Update app.py

app.py

@@ -31,7 +31,7 @@ def predict(img):
 
     # Reading the image and size transformation
     features = Image.open(img)
-    img = auto_transform(features).unsqueeze(0)
+    img = transforms(features).unsqueeze(0)
 
     with torch.inference_mode():
       output = model_alexnet(img)