Spaces:
Runtime error
Runtime error
Commit
·
4c665dd
1
Parent(s):
4e65042
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
|
|
| 18 |
# img_ = img_.add(torch.FloatTensor(MEAN).view(3, 1, 1)).detach().numpy()
|
| 19 |
# img_ = np.transpose(img_, (1, 2, 0))
|
| 20 |
# return img_
|
| 21 |
-
inverse_transform = transforms.Compose([
|
| 22 |
transforms.ToPILImage()
|
| 23 |
])
|
| 24 |
def predict(img):
|
|
|
|
| 18 |
# img_ = img_.add(torch.FloatTensor(MEAN).view(3, 1, 1)).detach().numpy()
|
| 19 |
# img_ = np.transpose(img_, (1, 2, 0))
|
| 20 |
# return img_
|
| 21 |
+
inverse_transform = transforms.Compose([ transforms.Normalize(mean=[-1.0, -1.0, -1.0], std=[2.0, 2.0, 2.0]),
|
| 22 |
transforms.ToPILImage()
|
| 23 |
])
|
| 24 |
def predict(img):
|