XiaRho commited on
Commit
b35d1f7
1 Parent(s): d50307c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ model_choice = {
20
  for model_type in model_choice.keys():
21
  model_choice[model_type] = hf_hub_download(repo_id="XiaRho/SEMat", filename=f"SEMat_{model_type}.pth", repo_type="model")
22
 
23
- def load_model(model_type='SAM2'):
24
  assert model_type in model_choice.keys()
25
  config_path = './configs/SEMat_{}.py'.format(model_type)
26
  cfg = LazyConfig.load(config_path)
@@ -120,7 +120,7 @@ if __name__ == '__main__':
120
 
121
  with gr.Row():
122
  with gr.Column(scale=45):
123
- input_model_type = gr.Dropdown(list(model_choice.keys()), value='SAM2', label="Trained SEMat Version")
124
 
125
  with gr.Column(scale=45):
126
  bt = gr.Button()
 
20
  for model_type in model_choice.keys():
21
  model_choice[model_type] = hf_hub_download(repo_id="XiaRho/SEMat", filename=f"SEMat_{model_type}.pth", repo_type="model")
22
 
23
+ def load_model(model_type='HQ-SAM'):
24
  assert model_type in model_choice.keys()
25
  config_path = './configs/SEMat_{}.py'.format(model_type)
26
  cfg = LazyConfig.load(config_path)
 
120
 
121
  with gr.Row():
122
  with gr.Column(scale=45):
123
+ input_model_type = gr.Dropdown(list(model_choice.keys()), value='HQ-SAM', label="Trained SEMat Version")
124
 
125
  with gr.Column(scale=45):
126
  bt = gr.Button()