Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,9 @@ def fig2img(fig):
|
|
| 61 |
def inference(
|
| 62 |
image_path: gr.inputs.Image = None,
|
| 63 |
model_path: gr.inputs.Dropdown = 'kadirnar/yolox_s-v0.1.1',
|
| 64 |
-
image_size: gr.inputs.Slider = 640
|
|
|
|
|
|
|
| 65 |
):
|
| 66 |
|
| 67 |
if model_name in ("yolox_s-v0.1.1", "yolox_m-v0.1.1", "yolox_tiny-v0.1.1"):
|
|
|
|
| 61 |
def inference(
|
| 62 |
image_path: gr.inputs.Image = None,
|
| 63 |
model_path: gr.inputs.Dropdown = 'kadirnar/yolox_s-v0.1.1',
|
| 64 |
+
image_size: gr.inputs.Slider = 640,
|
| 65 |
+
prob_threshold = 0.8,
|
| 66 |
+
"",
|
| 67 |
):
|
| 68 |
|
| 69 |
if model_name in ("yolox_s-v0.1.1", "yolox_m-v0.1.1", "yolox_tiny-v0.1.1"):
|