alexnasa commited on
Commit
2d034f7
·
verified ·
1 Parent(s): 697011b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -106,9 +106,11 @@ def run_triposg(image_path: str,
106
  num_tokens: int = 1024,
107
  num_inference_steps: int = 50,
108
  guidance_scale: float = 7.0,
109
- max_num_expanded_coords: str = 1e9,
110
  use_flash_decoder: bool = False,
111
  rmbg: bool = True):
 
 
 
112
  """
113
  Generate 3D part meshes from an input image.
114
  """
@@ -183,7 +185,6 @@ def build_demo():
183
  num_tokens = gr.Slider(256, 2048, value=1024, step=64, label="Num Tokens")
184
  num_steps = gr.Slider(1, 100, value=50, step=1, label="Inference Steps")
185
  guidance = gr.Slider(1.0, 20.0, value=7.0, step=0.1, label="Guidance Scale")
186
- max_coords = gr.Text(value="1e9", label="Max Expanded Coords")
187
  flash_decoder = gr.Checkbox(value=False, label="Use Flash Decoder")
188
  remove_bg = gr.Checkbox(value=False, label="Remove Background (RMBG)")
189
  run_button = gr.Button("Generate 3D Parts")
@@ -199,7 +200,6 @@ def build_demo():
199
  1024,
200
  50,
201
  7.0,
202
- 1e9,
203
  False,
204
  True
205
  ],
@@ -214,7 +214,7 @@ def build_demo():
214
 
215
  run_button.click(fn=run_triposg,
216
  inputs=[input_image, num_parts, seed, num_tokens, num_steps,
217
- guidance, max_coords, flash_decoder, remove_bg],
218
  outputs=[output_model, output_dir])
219
  return demo
220
 
 
106
  num_tokens: int = 1024,
107
  num_inference_steps: int = 50,
108
  guidance_scale: float = 7.0,
 
109
  use_flash_decoder: bool = False,
110
  rmbg: bool = True):
111
+
112
+ max_num_expanded_coords = 1e9
113
+
114
  """
115
  Generate 3D part meshes from an input image.
116
  """
 
185
  num_tokens = gr.Slider(256, 2048, value=1024, step=64, label="Num Tokens")
186
  num_steps = gr.Slider(1, 100, value=50, step=1, label="Inference Steps")
187
  guidance = gr.Slider(1.0, 20.0, value=7.0, step=0.1, label="Guidance Scale")
 
188
  flash_decoder = gr.Checkbox(value=False, label="Use Flash Decoder")
189
  remove_bg = gr.Checkbox(value=False, label="Remove Background (RMBG)")
190
  run_button = gr.Button("Generate 3D Parts")
 
200
  1024,
201
  50,
202
  7.0,
 
203
  False,
204
  True
205
  ],
 
214
 
215
  run_button.click(fn=run_triposg,
216
  inputs=[input_image, num_parts, seed, num_tokens, num_steps,
217
+ guidance, flash_decoder, remove_bg],
218
  outputs=[output_model, output_dir])
219
  return demo
220