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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -9
app.py CHANGED
@@ -106,21 +106,13 @@ 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: float = 1e9,
110
  use_flash_decoder: bool = False,
111
  rmbg: bool = True):
112
  """
113
  Generate 3D part meshes from an input image.
114
  """
115
 
116
- num_parts = 3
117
- seed = 0
118
- num_tokens = 1024
119
- num_inference_steps = 50
120
- guidance_scale = 7
121
- max_num_expanded_coords = 1e9
122
- use_flash_decoder = False
123
-
124
  if rmbg:
125
  img_pil = prepare_image(image_path, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
126
  else:
 
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
  """
115
 
 
 
 
 
 
 
 
 
116
  if rmbg:
117
  img_pil = prepare_image(image_path, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
118
  else: