michaelj commited on
Commit
3aafe99
·
verified ·
1 Parent(s): d7d97a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -10
app.py CHANGED
@@ -68,9 +68,9 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
68
  return image
69
 
70
 
71
- def generate(image, mc_resolution):
72
  scene_codes = model(image, device=device)
73
- mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
74
  mesh = to_gradio_3d_orientation(mesh)
75
  mesh_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False)
76
  mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
@@ -108,13 +108,6 @@ with gr.Blocks() as demo:
108
  value=0.85,
109
  step=0.05,
110
  )
111
- mc_resolution = gr.Slider(
112
- label="Marching Cubes Resolution",
113
- minimum=32,
114
- maximum=1024,
115
- value=256,
116
- step=32
117
- )
118
  with gr.Row():
119
  submit = gr.Button("Generate", elem_id="generate", variant="primary")
120
  with gr.Column():
@@ -146,7 +139,7 @@ with gr.Blocks() as demo:
146
  outputs=[processed_image],
147
  ).success(
148
  fn=generate,
149
- inputs=[processed_image, mc_resolution],
150
  outputs=[output_model, output_model2],
151
  )
152
 
 
68
  return image
69
 
70
 
71
+ def generate(image):
72
  scene_codes = model(image, device=device)
73
+ mesh = model.extract_mesh(scene_codes)[0]
74
  mesh = to_gradio_3d_orientation(mesh)
75
  mesh_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False)
76
  mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
 
108
  value=0.85,
109
  step=0.05,
110
  )
 
 
 
 
 
 
 
111
  with gr.Row():
112
  submit = gr.Button("Generate", elem_id="generate", variant="primary")
113
  with gr.Column():
 
139
  outputs=[processed_image],
140
  ).success(
141
  fn=generate,
142
+ inputs=[processed_image],
143
  outputs=[output_model, output_model2],
144
  )
145