Spaces:
Runtime error
Runtime error
Commit
·
71a3f83
1
Parent(s):
f79a186
Update app.py
Browse files
app.py
CHANGED
@@ -93,7 +93,7 @@ model2.eval()
|
|
93 |
|
94 |
def predict(input_img, ver):
|
95 |
input_img = Image.open(input_img)
|
96 |
-
transform = transforms.Compose([transforms.Resize(
|
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(512, Image.BICUBIC), transforms.ToTensor()])
|
97 |
input_img = transform(input_img)
|
98 |
input_img = torch.unsqueeze(input_img, 0)
|
99 |
|