FluoGen group

app.py

@@ -74,6 +74,8 @@ CONTROLNET_UNET_PATH = f"{MODELS_ROOT_DIR}/UNET_T2I_CONTROLNET/checkpoint-285000
 
 # Super-Resolution Models
 SR_CONTROLNET_MODELS = {
+    "Checkpoint ER": f"{MODELS_ROOT_DIR}/ControlNet_SR/ER/checkpoint-30000",
+    "Checkpoint Microtubules": f"{MODELS_ROOT_DIR}/ControlNet_SR/Microtubules/checkpoint-72500",
     "Checkpoint CCPs": f"{MODELS_ROOT_DIR}/ControlNet_SR/CCPs/checkpoint-100000",
     "Checkpoint F-actin": f"{MODELS_ROOT_DIR}/ControlNet_SR/F-actin/checkpoint-35000",
 }
@@ -143,6 +145,10 @@ def numpy_to_pil(image_np, target_mode="RGB"):
     return pil_image
 
 def update_sr_prompt(model_name):
+    if model_name == "Checkpoint ER":
+        return "ER of COS-7"
+    if model_name == "Checkpoint Microtubules":
+        return "Microtubules of COS-7"
     if model_name == "Checkpoint CCPs":
         return "CCPs of COS-7"
     elif model_name == "Checkpoint F-actin":