update

app.py

@@ -13,6 +13,10 @@ import spaces
 _TITLE = 'Gradio Demo of ScaleLSD for Structured Representation of Images'
 MAX_SEED = 1000
 
+os.system('mkdir -p models')
+os.system('wget https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v2-train-sa1b.pt -O models/scalelsd-vitbase-v2-train-sa1b.pt')
+os.system('wget https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v1-train-sa1b.pt -O models/scalelsd-vitbase-v1-train-sa1b.pt')
+
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     """random seed"""