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,
|