yuhe6 commited on
Commit
c7e3a64
·
1 Parent(s): 50bf796

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,9 +24,9 @@ torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images
24
  def inference(input_image):
25
  preprocess = transforms.Compose([
26
  transforms.Resize(256),
27
- transforms.CenterCrop(224),
28
  transforms.ToTensor(),
29
- transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
30
  ])
31
  input_tensor = preprocess(input_image)
32
  input_batch = input_tensor.unsqueeze(0) # create a mini-batch as expected by the model
 
24
  def inference(input_image):
25
  preprocess = transforms.Compose([
26
  transforms.Resize(256),
27
+ #transforms.CenterCrop(224),
28
  transforms.ToTensor(),
29
+ #transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
30
  ])
31
  input_tensor = preprocess(input_image)
32
  input_batch = input_tensor.unsqueeze(0) # create a mini-batch as expected by the model