Spaces:
Runtime error
Runtime error
Maternetworkin87
commited on
Commit
•
253c98d
1
Parent(s):
7da862c
Update model.py
Browse files
model.py
CHANGED
@@ -36,8 +36,8 @@ def create_effnetb2_model(num_classes:int=100,
|
|
36 |
|
37 |
return model
|
38 |
|
39 |
-
def manual_transforms(img)
|
40 |
-
transforms.Compose([
|
41 |
transforms.Resize((224, 224)), # 1. Reshape all images to 224x224 (though some models may require different sizes)
|
42 |
transforms.ToTensor(), # 2. Turn image values to between 0 & 1
|
43 |
transforms.Normalize(mean=[0.485, 0.456, 0.406], # 3. A mean of [0.485, 0.456, 0.406] (across each colour channel)
|
|
|
36 |
|
37 |
return model
|
38 |
|
39 |
+
def manual_transforms(img):
|
40 |
+
img = transforms.Compose([
|
41 |
transforms.Resize((224, 224)), # 1. Reshape all images to 224x224 (though some models may require different sizes)
|
42 |
transforms.ToTensor(), # 2. Turn image values to between 0 & 1
|
43 |
transforms.Normalize(mean=[0.485, 0.456, 0.406], # 3. A mean of [0.485, 0.456, 0.406] (across each colour channel)
|