Spaces:

cherubicxn
/

ScaleLSD

Running on Zero

Nan Xue commited on Jun 13

Commit

f5bdb68

1 Parent(s): 5b2c05c

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -169,7 +169,7 @@ def run_demo():
                 with gr.Row():
                     model_name = gr.Dropdown(
                         [ckpt for ckpt in os.listdir('models') if ckpt.endswith('.pt')],
-                        value='scalelsd-vitbase-v1-train-sa1b.pt',
                         label="Model Selection"
                     )

                 with gr.Row():
                     model_name = gr.Dropdown(
                         [ckpt for ckpt in os.listdir('models') if ckpt.endswith('.pt')],
+                        value='scalelsd-vitbase-v2-train-sa1b.pt',
                         label="Model Selection"
                     )