carolineec commited on
Commit
bd4b429
1 Parent(s): aef0baa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -93,7 +93,8 @@ model2.eval()
93
 
94
  def predict(input_img, ver):
95
  input_img = Image.open(input_img)
96
- transform = transforms.Compose([transforms.Resize(256, Image.BICUBIC), transforms.ToTensor()])
 
97
  input_img = transform(input_img)
98
  input_img = torch.unsqueeze(input_img, 0)
99
 
93
 
94
  def predict(input_img, ver):
95
  input_img = Image.open(input_img)
96
+ # transform = transforms.Compose([transforms.Resize(256, Image.BICUBIC), transforms.ToTensor()])
97
+ transform = transforms.Compose([transforms.ToTensor()])
98
  input_img = transform(input_img)
99
  input_img = torch.unsqueeze(input_img, 0)
100