huzey commited on
Commit
8ef0473
1 Parent(s): 4d49316

moved backbone models

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -585,7 +585,7 @@ def make_output_images_section():
585
 
586
  def make_parameters_section():
587
  gr.Markdown("### Parameters <a style='color: #0044CC;' href='https://ncut-pytorch.readthedocs.io/en/latest/how_to_get_better_segmentation/' target='_blank'>Help</a>")
588
- from backbone import get_demo_model_names
589
  model_names = get_demo_model_names()
590
  model_dropdown = gr.Dropdown(model_names, label="Backbone", value="DiNO(dino_vitb8)", elem_id="model_name")
591
  layer_slider = gr.Slider(1, 12, step=1, label="Backbone: Layer index", value=10, elem_id="layer")
 
585
 
586
  def make_parameters_section():
587
  gr.Markdown("### Parameters <a style='color: #0044CC;' href='https://ncut-pytorch.readthedocs.io/en/latest/how_to_get_better_segmentation/' target='_blank'>Help</a>")
588
+ from ncut_pytorch.backbone import get_demo_model_names
589
  model_names = get_demo_model_names()
590
  model_dropdown = gr.Dropdown(model_names, label="Backbone", value="DiNO(dino_vitb8)", elem_id="model_name")
591
  layer_slider = gr.Slider(1, 12, step=1, label="Backbone: Layer index", value=10, elem_id="layer")