Spaces:
Sleeping
Sleeping
Commit ·
9d79386
1
Parent(s): 742adc7
fixed model state path
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(num_classes=3)
|
|
| 16 |
|
| 17 |
# Load saved weights
|
| 18 |
|
| 19 |
-
effnetb2.load_state_dict(torch.load(f="
|
| 20 |
map_location=torch.device("cpu")))
|
| 21 |
|
| 22 |
### 3. Predict function ###
|
|
@@ -60,4 +60,4 @@ demo = gr.Interface(fn=predict, # maps inputs to outputs
|
|
| 60 |
description=description,
|
| 61 |
article=article)
|
| 62 |
# launch the demo
|
| 63 |
-
demo.launch()
|
|
|
|
| 16 |
|
| 17 |
# Load saved weights
|
| 18 |
|
| 19 |
+
effnetb2.load_state_dict(torch.load(f="pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
|
| 20 |
map_location=torch.device("cpu")))
|
| 21 |
|
| 22 |
### 3. Predict function ###
|
|
|
|
| 60 |
description=description,
|
| 61 |
article=article)
|
| 62 |
# launch the demo
|
| 63 |
+
demo.launch(debug=False)
|