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([
|