CharlieAmalet commited on
Commit
8f8d235
·
verified ·
1 Parent(s): c5c5a80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -114,16 +114,16 @@ with gr.Blocks(css=css) as API:
114
  gr.Markdown(description)
115
  with gr.Tab("Depth Prediction"):
116
  with gr.Row():
117
- inputs=gr.Image(label="Input Image", type='pil') # Input is an image
118
- outputs=gr.Image(label="Depth Map", type='pil') # Output is also an image
119
  generate_btn = gr.Button(value="Generate")
120
  generate_btn.click(partial(process_image, model), inputs=inputs, outputs=outputs, api_name="generate_depth")
121
 
122
  with gr.Tab("Image to 3D"):
123
  with gr.Row():
124
  with gr.Column():
125
- inputs=[gr.Image(label="Input Image", type='pil'), gr.Checkbox(label="Keep occlusion edges", value=True)]
126
- outputs=gr.Model3D(label="3D Mesh", clear_color=[1.0, 1.0, 1.0, 1.0])
127
  generate_btn = gr.Button(value="Generate")
128
  generate_btn.click(partial(get_mesh, model), inputs=inputs, outputs=outputs, api_name="generate_mesh")
129
 
 
114
  gr.Markdown(description)
115
  with gr.Tab("Depth Prediction"):
116
  with gr.Row():
117
+ inputs=gr.Image(label="Input Image", type='pil', height=500) # Input is an image
118
+ outputs=gr.Image(label="Depth Map", type='pil', height=500) # Output is also an image
119
  generate_btn = gr.Button(value="Generate")
120
  generate_btn.click(partial(process_image, model), inputs=inputs, outputs=outputs, api_name="generate_depth")
121
 
122
  with gr.Tab("Image to 3D"):
123
  with gr.Row():
124
  with gr.Column():
125
+ inputs=[gr.Image(label="Input Image", type='pil', height=500), gr.Checkbox(label="Keep occlusion edges", value=True)]
126
+ outputs=gr.Model3D(label="3D Mesh", clear_color=[1.0, 1.0, 1.0, 1.0], height=500)
127
  generate_btn = gr.Button(value="Generate")
128
  generate_btn.click(partial(get_mesh, model), inputs=inputs, outputs=outputs, api_name="generate_mesh")
129