Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|