Spaces:
Sleeping
Sleeping
more prompt
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ M2I_CONTROLNET_PATH = f"{MODELS_ROOT_DIR}/ControlNet_M2I/checkpoint-30000"
|
|
| 63 |
M2I_EXAMPLE_IMG_DIR = "example_images_m2i"
|
| 64 |
|
| 65 |
# --- Tab 2: Text-to-Image Config ---
|
| 66 |
-
T2I_PROMPTS = ["F-actin of COS-7", "ER of COS-7", "Mitochondria of BPAE", "Nucleus of BPAE", "ER of HeLa", "Microtubules of HeLa"]
|
| 67 |
T2I_EXAMPLE_IMG_DIR = "example_images"
|
| 68 |
T2I_CHECKPOINT = 285000
|
| 69 |
T2I_PRETRAINED_MODEL_PATH = f"{MODELS_ROOT_DIR}/stable-diffusion-v1-5"
|
|
@@ -151,6 +151,7 @@ def load_all_prompts():
|
|
| 151 |
print(f"Error loading prompts: {e}")
|
| 152 |
# 如果加载失败,返回几个默认值保底
|
| 153 |
return ["F-actin of COS-7", "ER of COS-7", "Mitochondria of BPAE"]
|
|
|
|
| 154 |
|
| 155 |
# --- 1. Model Loading ---
|
| 156 |
print("--- Initializing FluoGen Application ---")
|
|
|
|
| 63 |
M2I_EXAMPLE_IMG_DIR = "example_images_m2i"
|
| 64 |
|
| 65 |
# --- Tab 2: Text-to-Image Config ---
|
| 66 |
+
# T2I_PROMPTS = ["F-actin of COS-7", "ER of COS-7", "Mitochondria of BPAE", "Nucleus of BPAE", "ER of HeLa", "Microtubules of HeLa"]
|
| 67 |
T2I_EXAMPLE_IMG_DIR = "example_images"
|
| 68 |
T2I_CHECKPOINT = 285000
|
| 69 |
T2I_PRETRAINED_MODEL_PATH = f"{MODELS_ROOT_DIR}/stable-diffusion-v1-5"
|
|
|
|
| 151 |
print(f"Error loading prompts: {e}")
|
| 152 |
# 如果加载失败,返回几个默认值保底
|
| 153 |
return ["F-actin of COS-7", "ER of COS-7", "Mitochondria of BPAE"]
|
| 154 |
+
T2I_PROMPTS = load_all_prompts()
|
| 155 |
|
| 156 |
# --- 1. Model Loading ---
|
| 157 |
print("--- Initializing FluoGen Application ---")
|