Added image titles to the three viewers
Browse files- demo/src/gui.py +4 -4
demo/src/gui.py
CHANGED
|
@@ -20,7 +20,7 @@ class WebUI:
|
|
| 20 |
self.pred_images = []
|
| 21 |
|
| 22 |
# @TODO: This should be dynamically set based on chosen volume size
|
| 23 |
-
self.nb_slider_items =
|
| 24 |
|
| 25 |
self.model_name = model_name
|
| 26 |
self.cwd = cwd
|
|
@@ -151,7 +151,7 @@ class WebUI:
|
|
| 151 |
for i in range(self.nb_slider_items):
|
| 152 |
visibility = True if i == 1 else False
|
| 153 |
t = gr.Image(
|
| 154 |
-
visible=visibility, elem_id="model-2d-fixed"
|
| 155 |
).style(
|
| 156 |
height=512,
|
| 157 |
width=512,
|
|
@@ -162,7 +162,7 @@ class WebUI:
|
|
| 162 |
for i in range(self.nb_slider_items):
|
| 163 |
visibility = True if i == 1 else False
|
| 164 |
t = gr.Image(
|
| 165 |
-
visible=visibility, elem_id="model-2d-moving"
|
| 166 |
).style(
|
| 167 |
height=512,
|
| 168 |
width=512,
|
|
@@ -173,7 +173,7 @@ class WebUI:
|
|
| 173 |
for i in range(self.nb_slider_items):
|
| 174 |
visibility = True if i == 1 else False
|
| 175 |
t = gr.Image(
|
| 176 |
-
visible=visibility, elem_id="model-2d-pred"
|
| 177 |
).style(
|
| 178 |
height=512,
|
| 179 |
width=512,
|
|
|
|
| 20 |
self.pred_images = []
|
| 21 |
|
| 22 |
# @TODO: This should be dynamically set based on chosen volume size
|
| 23 |
+
self.nb_slider_items = 128
|
| 24 |
|
| 25 |
self.model_name = model_name
|
| 26 |
self.cwd = cwd
|
|
|
|
| 151 |
for i in range(self.nb_slider_items):
|
| 152 |
visibility = True if i == 1 else False
|
| 153 |
t = gr.Image(
|
| 154 |
+
visible=visibility, elem_id="model-2d-fixed", label="fixed image", show_label=True,
|
| 155 |
).style(
|
| 156 |
height=512,
|
| 157 |
width=512,
|
|
|
|
| 162 |
for i in range(self.nb_slider_items):
|
| 163 |
visibility = True if i == 1 else False
|
| 164 |
t = gr.Image(
|
| 165 |
+
visible=visibility, elem_id="model-2d-moving", label="moving image", show_label=True,
|
| 166 |
).style(
|
| 167 |
height=512,
|
| 168 |
width=512,
|
|
|
|
| 173 |
for i in range(self.nb_slider_items):
|
| 174 |
visibility = True if i == 1 else False
|
| 175 |
t = gr.Image(
|
| 176 |
+
visible=visibility, elem_id="model-2d-pred", label="predicted fixed image", show_label=True,
|
| 177 |
).style(
|
| 178 |
height=512,
|
| 179 |
width=512,
|