Update app.py

app.py

@@ -93,8 +93,8 @@ model2.eval()
 
 def predict(input_img, ver):
     input_img = Image.open(input_img)
-   # transform = transforms.Compose([transforms.Resize(1024, Image.BICUBIC), transforms.ToTensor()])
-   # input_img = transform(input_img)
+    transform = transforms.Compose([transforms.Resize(1024, Image.BICUBIC), transforms.ToTensor()])
+    input_img = transform(input_img)
     input_img = torch.unsqueeze(input_img, 0)
 
     drawing = 0