andreped commited on
Commit
ba66955
·
unverified ·
1 Parent(s): d3ab234

Fix for Upload multiselection in demo

Browse files
Files changed (1) hide show
  1. demo/src/gui.py +23 -4
demo/src/gui.py CHANGED
@@ -45,8 +45,8 @@ class WebUI:
45
  print("Changed task to:", value)
46
  self.class_name = value
47
 
48
- def upload_file(self, file):
49
- return file.name
50
 
51
  def process(self, mesh_file_names):
52
  fixed_image_path = mesh_file_names[0].name
@@ -150,8 +150,27 @@ class WebUI:
150
  )
151
  fixed_images.append(t)
152
 
153
- moving_images = fixed_images.copy()
154
- pred_images = fixed_images.copy()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
155
 
156
  self.slider.input(
157
  self.get_fixed_image, self.slider, fixed_images
 
45
  print("Changed task to:", value)
46
  self.class_name = value
47
 
48
+ def upload_file(self, files):
49
+ return files
50
 
51
  def process(self, mesh_file_names):
52
  fixed_image_path = mesh_file_names[0].name
 
150
  )
151
  fixed_images.append(t)
152
 
153
+ moving_images = []
154
+ for i in range(self.nb_slider_items):
155
+ visibility = True if i == 1 else False
156
+ t = gr.Image(
157
+ visible=visibility, elem_id="model-2d"
158
+ ).style(
159
+ height=512,
160
+ width=512,
161
+ )
162
+ moving_images.append(t)
163
+
164
+ pred_images = []
165
+ for i in range(self.nb_slider_items):
166
+ visibility = True if i == 1 else False
167
+ t = gr.Image(
168
+ visible=visibility, elem_id="model-2d"
169
+ ).style(
170
+ height=512,
171
+ width=512,
172
+ )
173
+ pred_images.append(t)
174
 
175
  self.slider.input(
176
  self.get_fixed_image, self.slider, fixed_images