Update src/streamlit_app.py

src/streamlit_app.py

@@ -43,14 +43,15 @@ seed = st.number_input("🎲 Random Seed (-1 for random)", value=-1, step=1)
 size_level = st.number_input("📐 Size Level (minimum 512)", value=512, min_value=512, step=32)
 generate_button = st.button("🚀 Generate")
 
-# === Cached Model Loader ===
+
 @st.cache_resource
 def load_model():
     repo = "stepfun-ai/Step1X-Edit"
     local_dir = Path.home() / "step1x_weights"
     local_dir.mkdir(exist_ok=True)
 
-    snapshot_download(repo_id=repo, local_dir=local_dir, local_dir_use_symlinks=False)
+    # Removed deprecated `local_dir_use_symlinks` parameter
+    snapshot_download(repo_id=repo, local_dir=local_dir)
 
     model = ImageGenerator(
         ae_path=local_dir / 'vae.safetensors',
@@ -60,6 +61,7 @@ def load_model():
     )
     return model
 
+
 image_edit_model = load_model()
 
 # === Handle Generation ===