czkaiweb commited on
Commit
e1d5e6c
·
1 Parent(s): 0a777bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def inference(input_image):
26
  transforms.Resize(260),
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
 
26
  transforms.Resize(260),
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