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)
|