Hpsoyl commited on
Commit
d7f65f5
·
1 Parent(s): 9355932
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -75,7 +75,7 @@ CONTROLNET_UNET_PATH = f"{MODELS_ROOT_DIR}/UNET_T2I_CONTROLNET/checkpoint-285000
75
 
76
  SR_CONTROLNET_MODELS = {
77
  "Checkpoint ER": f"{MODELS_ROOT_DIR}/ControlNet_SR/ER/checkpoint-30000",
78
- "Checkpoint Microbules": f"{MODELS_ROOT_DIR}/ControlNet_SR/Microtubules/checkpoint-72500",
79
  "Checkpoint CCPs": f"{MODELS_ROOT_DIR}/ControlNet_SR/CCPs/checkpoint-100000",
80
  "Checkpoint F-actin": f"{MODELS_ROOT_DIR}/ControlNet_SR/F-actin/checkpoint-35000",
81
  }
@@ -639,7 +639,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
639
  sr_out_disp = gr.Image(label="Output", type="pil", interactive=False, show_download_button=False)
640
  with gr.Row(equal_height=True):
641
  with gr.Column(scale=2):
642
- sr_color = gr.Dropdown(choices=COLOR_MAPS, value="Red (RFP)", label="Pseudocolor")
643
  with gr.Column(scale=2):
644
  sr_colorbar = gr.Image(label="Colorbar", show_label=False, container=False, height=40, show_download_button=False, interactive=False)
645
  with gr.Column(scale=1):
 
75
 
76
  SR_CONTROLNET_MODELS = {
77
  "Checkpoint ER": f"{MODELS_ROOT_DIR}/ControlNet_SR/ER/checkpoint-30000",
78
+ "Checkpoint Microtubules": f"{MODELS_ROOT_DIR}/ControlNet_SR/Microtubules/checkpoint-72500",
79
  "Checkpoint CCPs": f"{MODELS_ROOT_DIR}/ControlNet_SR/CCPs/checkpoint-100000",
80
  "Checkpoint F-actin": f"{MODELS_ROOT_DIR}/ControlNet_SR/F-actin/checkpoint-35000",
81
  }
 
639
  sr_out_disp = gr.Image(label="Output", type="pil", interactive=False, show_download_button=False)
640
  with gr.Row(equal_height=True):
641
  with gr.Column(scale=2):
642
+ sr_color = gr.Dropdown(choices=COLOR_MAPS, value="Red Hot", label="Pseudocolor")
643
  with gr.Column(scale=2):
644
  sr_colorbar = gr.Image(label="Colorbar", show_label=False, container=False, height=40, show_download_button=False, interactive=False)
645
  with gr.Column(scale=1):