Spaces:
Running
on
Zero
Running
on
Zero
Improve app
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ MODEL_VERSIONS = {
|
|
30 |
"v1.8.1: Best Sharpness": "yoso-normal-v1-8-1"
|
31 |
}
|
32 |
|
33 |
-
def load_predictor(version: str = "v1.8.1"):
|
34 |
"""Load model predictor using torch.hub with specified version"""
|
35 |
if version not in predictors:
|
36 |
yoso_version = MODEL_VERSIONS[version]
|
@@ -55,7 +55,7 @@ def precache_all_predictors():
|
|
55 |
|
56 |
def process_image(
|
57 |
path_input: str,
|
58 |
-
version: str = "v1.8.1",
|
59 |
data_type: str = "object"
|
60 |
) -> tuple:
|
61 |
"""Process single image with specified model version"""
|
@@ -128,7 +128,7 @@ def create_demo():
|
|
128 |
# Model version selector
|
129 |
version_dropdown = gr.Dropdown(
|
130 |
choices=list(MODEL_VERSIONS.keys()),
|
131 |
-
value="v1.8.1",
|
132 |
label="Model Version",
|
133 |
info="Select YOSO Normal model version"
|
134 |
)
|
@@ -176,7 +176,7 @@ def create_demo():
|
|
176 |
)
|
177 |
|
178 |
object_reset_btn.click(
|
179 |
-
fn=lambda: (None, "v1.8.1", None),
|
180 |
inputs=[],
|
181 |
outputs=[object_input, version_dropdown, object_output_slider],
|
182 |
queue=False,
|
|
|
30 |
"v1.8.1: Best Sharpness": "yoso-normal-v1-8-1"
|
31 |
}
|
32 |
|
33 |
+
def load_predictor(version: str = "v1.8.1: Best Sharpness"):
|
34 |
"""Load model predictor using torch.hub with specified version"""
|
35 |
if version not in predictors:
|
36 |
yoso_version = MODEL_VERSIONS[version]
|
|
|
55 |
|
56 |
def process_image(
|
57 |
path_input: str,
|
58 |
+
version: str = "v1.8.1: Best Sharpness",
|
59 |
data_type: str = "object"
|
60 |
) -> tuple:
|
61 |
"""Process single image with specified model version"""
|
|
|
128 |
# Model version selector
|
129 |
version_dropdown = gr.Dropdown(
|
130 |
choices=list(MODEL_VERSIONS.keys()),
|
131 |
+
value="v1.8.1: Best Sharpness",
|
132 |
label="Model Version",
|
133 |
info="Select YOSO Normal model version"
|
134 |
)
|
|
|
176 |
)
|
177 |
|
178 |
object_reset_btn.click(
|
179 |
+
fn=lambda: (None, "v1.8.1: Best Sharpness", None),
|
180 |
inputs=[],
|
181 |
outputs=[object_input, version_dropdown, object_output_slider],
|
182 |
queue=False,
|