kfahn commited on
Commit
f1d03bc
·
1 Parent(s): 5e9da35

Update app.py

Browse files

remove version

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ model2 = Generator(3, 1, 3)
91
  model2.load_state_dict(torch.load('model2.pth', map_location=torch.device('cpu')))
92
  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)
 
91
  model2.load_state_dict(torch.load('model2.pth', map_location=torch.device('cpu')))
92
  model2.eval()
93
 
94
+ def predict(input_img):
95
  input_img = Image.open(input_img)
96
  transform = transforms.Compose([transforms.Resize(256, Image.BICUBIC), transforms.ToTensor()])
97
  input_img = transform(input_img)