Update app.py

app.py

@@ -39,7 +39,7 @@ transform = transforms.Compose(
 
 def predict(img):
     img = transform(img).to('cpu')
-    img = transform(img).unsqueeze(0)
+    img = img.unsqueeze(0)
     with torch.no_grad():
         out=  model(img)
     probability = torch.nn.functional.softmax(out[0],dim=0)