adpro commited on
Commit
047c406
·
1 Parent(s): b96f071

Update app.py

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