Spaces:
Sleeping
Sleeping
color
Browse files
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
|
| 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
|
| 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):
|