alexnasa commited on
Commit
758e6f6
·
verified ·
1 Parent(s): 2d1dbb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -187,7 +187,7 @@ def run_triposg(image_path: str,
187
 
188
  mesh_file = first_file_from_dir(export_dir, "glb")
189
 
190
- return mesh_file, export_dir, parts
191
 
192
  def cleanup(request: gr.Request):
193
 
@@ -272,7 +272,7 @@ def build_demo():
272
  ],
273
  ],
274
  inputs=[input_image, num_parts],
275
- outputs=[output_model, output_dir, parts_model],
276
  fn=run_triposg,
277
  cache_examples=True,
278
  )
@@ -280,7 +280,7 @@ def build_demo():
280
  run_button.click(fn=run_triposg,
281
  inputs=[input_image, num_parts, seed, num_tokens, num_steps,
282
  guidance, flash_decoder, remove_bg, session_state],
283
- outputs=[output_model, output_dir, parts_model])
284
  return demo
285
 
286
  if __name__ == "__main__":
 
187
 
188
  mesh_file = first_file_from_dir(export_dir, "glb")
189
 
190
+ return mesh_file, export_dir
191
 
192
  def cleanup(request: gr.Request):
193
 
 
272
  ],
273
  ],
274
  inputs=[input_image, num_parts],
275
+ outputs=[output_model, output_dir],
276
  fn=run_triposg,
277
  cache_examples=True,
278
  )
 
280
  run_button.click(fn=run_triposg,
281
  inputs=[input_image, num_parts, seed, num_tokens, num_steps,
282
  guidance, flash_decoder, remove_bg, session_state],
283
+ outputs=[output_model, output_dir])
284
  return demo
285
 
286
  if __name__ == "__main__":