Spaces:
Runtime error
Runtime error
Update app.py
Browse filesremove version
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
|
| 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)
|