Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ with tf.keras.utils.custom_object_scope({"KerasLayer": hub.KerasLayer}):
|
|
| 44 |
|
| 45 |
# Define preprocessing function
|
| 46 |
def preprocess_image(image):
|
| 47 |
-
|
| 48 |
transforms.Resize((224, 224)),
|
| 49 |
transforms.ToTensor(),
|
| 50 |
transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
|
|
|
|
| 44 |
|
| 45 |
# Define preprocessing function
|
| 46 |
def preprocess_image(image):
|
| 47 |
+
transform_test = transforms.Compose([
|
| 48 |
transforms.Resize((224, 224)),
|
| 49 |
transforms.ToTensor(),
|
| 50 |
transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
|