Spaces:
Sleeping
Sleeping
fixed flagging parameter
Browse files
app.py
CHANGED
|
@@ -51,7 +51,7 @@ def predict_image(array):
|
|
| 51 |
input_image = Image.fromarray(array)
|
| 52 |
|
| 53 |
# Load the model
|
| 54 |
-
model = load_efficientnet_model('
|
| 55 |
|
| 56 |
# Transform the image
|
| 57 |
transform = transforms.Compose([
|
|
|
|
| 51 |
input_image = Image.fromarray(array)
|
| 52 |
|
| 53 |
# Load the model
|
| 54 |
+
model = load_efficientnet_model('efficientnet_epoch=18_loss=0.0020_val_f1score=0.8993.pth')
|
| 55 |
|
| 56 |
# Transform the image
|
| 57 |
transform = transforms.Compose([
|