Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ model_base = "https://huggingface.co/mayrajeo/marine-vessel-detection/resolve/ma
|
|
| 14 |
|
| 15 |
def inference(
|
| 16 |
im:gr.inputs.Image=None,
|
| 17 |
-
model_path:gr.inputs.Dropdown=
|
| 18 |
conf_thr:gr.inputs.Slider=0.25
|
| 19 |
):
|
| 20 |
model = Yolov8DetectionModel(model_path=f'{model_base}/{model_path}/best.pt',
|
|
|
|
| 14 |
|
| 15 |
def inference(
|
| 16 |
im:gr.inputs.Image=None,
|
| 17 |
+
model_path:gr.inputs.Dropdown='YOLOv8n',
|
| 18 |
conf_thr:gr.inputs.Slider=0.25
|
| 19 |
):
|
| 20 |
model = Yolov8DetectionModel(model_path=f'{model_base}/{model_path}/best.pt',
|