Spaces:
Sleeping
Sleeping
Commit
·
31b3106
1
Parent(s):
fa8be74
debug5
Browse files
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
|
| 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 =
|
| 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) ---
|