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