Harsimran19 commited on
Commit
4c665dd
·
1 Parent(s): 4e65042

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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([ transforms.Resize((64, 64)),transforms.Normalize(mean=[-1.0, -1.0, -1.0], std=[2.0, 2.0, 2.0]),
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):