Update app.py
Browse files
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='
|
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='
|
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()
|