Hpsoyl commited on
Commit
7f3f28e
·
1 Parent(s): 4610725

more prompt

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 ---")