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