Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -646,7 +646,7 @@ num_classes = len(dog_breeds)
|
|
| 646 |
model = BaseModel(num_classes=num_classes, device=device)
|
| 647 |
|
| 648 |
# Load model path
|
| 649 |
-
model_path = '
|
| 650 |
checkpoint = torch.load(model_path, map_location=device)
|
| 651 |
|
| 652 |
# Load model state
|
|
@@ -964,11 +964,11 @@ def main():
|
|
| 964 |
with gr.Tabs():
|
| 965 |
# 1. 品種檢測標籤頁
|
| 966 |
example_images = [
|
| 967 |
-
'
|
| 968 |
-
'
|
| 969 |
-
'
|
| 970 |
-
'
|
| 971 |
-
'
|
| 972 |
]
|
| 973 |
detection_components = create_detection_tab(predict, example_images)
|
| 974 |
|
|
|
|
| 646 |
model = BaseModel(num_classes=num_classes, device=device)
|
| 647 |
|
| 648 |
# Load model path
|
| 649 |
+
model_path = '[124_82.30]_best_model_dog.pth'
|
| 650 |
checkpoint = torch.load(model_path, map_location=device)
|
| 651 |
|
| 652 |
# Load model state
|
|
|
|
| 964 |
with gr.Tabs():
|
| 965 |
# 1. 品種檢測標籤頁
|
| 966 |
example_images = [
|
| 967 |
+
'Border_Collie.jpg',
|
| 968 |
+
'Golden_Retriever.jpeg',
|
| 969 |
+
'Saint_Bernard.jpeg',
|
| 970 |
+
'Samoyed.jpg',
|
| 971 |
+
'French_Bulldog.jpeg'
|
| 972 |
]
|
| 973 |
detection_components = create_detection_tab(predict, example_images)
|
| 974 |
|