Stable-X commited on
Commit
7caafe1
·
1 Parent(s): 0680865

Improve app

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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,