Spaces:
Sleeping
Sleeping
AlshimaaGamalAlsaied
commited on
Commit
•
c824d7a
1
Parent(s):
dfe8c33
update
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ from huggingface_hub import HfApi
|
|
14 |
|
15 |
def yolov7_inference(
|
16 |
image: gr.inputs.Image = None,
|
17 |
-
model_path: gr.inputs.Dropdown =
|
18 |
image_size: gr.inputs.Slider = 640,
|
19 |
conf_threshold: gr.inputs.Slider = 0.25,
|
20 |
iou_threshold: gr.inputs.Slider = 0.45,
|
@@ -42,10 +42,11 @@ inputs = [
|
|
42 |
gr.inputs.Image(type="pil", label="Input Image"),
|
43 |
gr.inputs.Dropdown(
|
44 |
choices=[
|
|
|
45 |
"kadirnar/yolov7-tiny-v0.1",
|
46 |
"kadirnar/yolov7-v0.1",
|
47 |
],
|
48 |
-
default="
|
49 |
label="Model",
|
50 |
),
|
51 |
gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
|
|
|
14 |
|
15 |
def yolov7_inference(
|
16 |
image: gr.inputs.Image = None,
|
17 |
+
model_path: gr.inputs.Dropdown = None,
|
18 |
image_size: gr.inputs.Slider = 640,
|
19 |
conf_threshold: gr.inputs.Slider = 0.25,
|
20 |
iou_threshold: gr.inputs.Slider = 0.45,
|
|
|
42 |
gr.inputs.Image(type="pil", label="Input Image"),
|
43 |
gr.inputs.Dropdown(
|
44 |
choices=[
|
45 |
+
"alshimaa/model_baseline"
|
46 |
"kadirnar/yolov7-tiny-v0.1",
|
47 |
"kadirnar/yolov7-v0.1",
|
48 |
],
|
49 |
+
default="alshimaa/model_baseline",
|
50 |
label="Model",
|
51 |
),
|
52 |
gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
|