rayquaza384mega commited on
Commit
31b3106
·
1 Parent(s): fa8be74
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,7 +34,7 @@ from models.pipeline_controlnet import DDPMControlnetPipeline
34
  # --- New Import for Segmentation ---
35
  from cellpose import models as cellpose_models
36
  from cellpose import plot as cellpose_plot
37
- from huggingface_hub import hf_hub_download
38
 
39
  # --- 0. Configuration & Constants ---
40
  # --- General ---
@@ -55,7 +55,7 @@ SAVE_EXAMPLES = False
55
  # NOTE: All model paths are now relative.
56
  # Run the `copy_weights.py` script once to copy all necessary model files into this local directory.
57
  REPO_ID = "rayquaza384mega/FluoGen-demo-test-ckpts"
58
- MODELS_ROOT_DIR = hf_hub_download(repo_id=REPO_ID) #"models_collection"
59
 
60
 
61
  # --- Tab 1: Mask-to-Image Config (Formerly Segmentation-to-Image) ---
 
34
  # --- New Import for Segmentation ---
35
  from cellpose import models as cellpose_models
36
  from cellpose import plot as cellpose_plot
37
+ from huggingface_hub import snapshot_download
38
 
39
  # --- 0. Configuration & Constants ---
40
  # --- General ---
 
55
  # NOTE: All model paths are now relative.
56
  # Run the `copy_weights.py` script once to copy all necessary model files into this local directory.
57
  REPO_ID = "rayquaza384mega/FluoGen-demo-test-ckpts"
58
+ MODELS_ROOT_DIR = snapshot_download(repo_id=REPO_ID) #"models_collection"
59
 
60
 
61
  # --- Tab 1: Mask-to-Image Config (Formerly Segmentation-to-Image) ---