FluoGen group

app.py

@@ -37,6 +37,7 @@ from cellpose import plot as cellpose_plot
 from huggingface_hub import snapshot_download
 
 # --- 0. Configuration & Constants ---
+hf_token = os.environ.get("HF_TOKEN")
 # --- General ---
 MODEL_TITLE = "🔬 FluoGen: AI-Powered Fluorescence Microscopy Suite"
 MODEL_DESCRIPTION = """
@@ -55,7 +56,7 @@ SAVE_EXAMPLES = False
 # NOTE: All model paths are now relative.
 # Run the `copy_weights.py` script once to copy all necessary model files into this local directory.
 REPO_ID = "FluoGen-Group/FluoGen-demo-test-ckpts"
-MODELS_ROOT_DIR = snapshot_download(repo_id=REPO_ID) #"models_collection"
+MODELS_ROOT_DIR = snapshot_download(repo_id=REPO_ID, token=hf_token) #"models_collection"
 
 # --- Tab 1: Mask-to-Image Config (Formerly Segmentation-to-Image) ---
 M2I_CONTROLNET_PATH = f"{MODELS_ROOT_DIR}/ControlNet_M2I/checkpoint-30000"