Update app.py

app.py

@@ -41,7 +41,7 @@ if uploaded_file is not None:
     
     cust_transform = A.Compose([A.Resize(height=256, width=256, p=1.0),ToTensorV2(p=1.0)], p=1.0)
     tensor = cust_transform(image=img)
-    tensor = tensor['image'].float(),resize(1,3,256,256)
+    tensor = tensor['image'].float().resize(1,3,256,256)
     
     custom_pred = model.forward(tensor).detach().numpy()
     custom_pred