gabsm commited on
Commit
cf75aba
1 Parent(s): 0fa8576

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,7 +56,7 @@ def predict(
56
  negative_prompt=negative_prompt,
57
  guidance_scale=guidance_scale,
58
  generator=generator,
59
- num_inference_steps=100,
60
  ) # type: ignore
61
  rgb_image, depth_image = output.rgb[0], output.depth[0] # type: ignore
62
  with NamedTemporaryFile(suffix=".png", delete=False, dir="tmp") as rgb_file:
@@ -101,7 +101,7 @@ Model card: https://huggingface.co/Intel/ldm3d-pano<br>
101
  depth = gr.Image(label="Depth Image", type="filepath")
102
  gr.Examples(
103
  examples=[
104
- ["360 view of a large bedroom", "", 7.0, 0, True]],
105
  inputs=[prompt, negative_prompt, guidance_scale, seed, randomize_seed],
106
  outputs=[rgb, depth, generated_seed, html],
107
  fn=predict,
 
56
  negative_prompt=negative_prompt,
57
  guidance_scale=guidance_scale,
58
  generator=generator,
59
+ num_inference_steps=50,
60
  ) # type: ignore
61
  rgb_image, depth_image = output.rgb[0], output.depth[0] # type: ignore
62
  with NamedTemporaryFile(suffix=".png", delete=False, dir="tmp") as rgb_file:
 
101
  depth = gr.Image(label="Depth Image", type="filepath")
102
  gr.Examples(
103
  examples=[
104
+ ["360 view of a large bedroom", "", 7.0, 24, False]],
105
  inputs=[prompt, negative_prompt, guidance_scale, seed, randomize_seed],
106
  outputs=[rgb, depth, generated_seed, html],
107
  fn=predict,