carolineec
commited on
Commit
•
bd4b429
1
Parent(s):
aef0baa
Update app.py
Browse files
app.py
CHANGED
@@ -93,7 +93,8 @@ 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)
|
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(256, Image.BICUBIC), transforms.ToTensor()])
|
97 |
+
transform = transforms.Compose([transforms.ToTensor()])
|
98 |
input_img = transform(input_img)
|
99 |
input_img = torch.unsqueeze(input_img, 0)
|
100 |
|