Update app.py

app.py

@@ -45,6 +45,7 @@ def predict(
     prompt: str,
     negative_prompt: str,
     guidance_scale: float = 5.0,
+    ddim_steps: int = 50,
     seed: int = 0,
     randomize_seed: bool = True,
 ):
@@ -55,6 +56,7 @@ def predict(
         height=512,
         negative_prompt=negative_prompt,
         guidance_scale=guidance_scale,
+        ddim_steps=ddim_steps,
         generator=generator,
         num_inference_steps=50,
     )  # type: ignore
@@ -87,6 +89,9 @@ Model card: https://huggingface.co/Intel/ldm3d-pano<br>
             guidance_scale = gr.Slider(
                 label="Guidance Scale", minimum=0, maximum=10, step=0.1, value=5.0
             )
+            ddim_steps = gr.Slider(
+                label="DDIM Steps", minimum=0, maximum=200, step=1, value=50
+            )
             randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
             seed = gr.Slider(label="Seed", minimum=0,
                              maximum=2**64 - 1, step=1)
@@ -101,15 +106,15 @@ Model card: https://huggingface.co/Intel/ldm3d-pano<br>
                 depth = gr.Image(label="Depth Image", type="filepath")
     gr.Examples(
         examples=[
-            ["360 view of a large bedroom", "", 7.0, 42, False]],
-        inputs=[prompt, negative_prompt, guidance_scale, seed, randomize_seed],
+            ["360 view of a large bedroom", "", 7.0, 50, 42, False]],
+        inputs=[prompt, negative_prompt, guidance_scale, ddim_steps, seed, randomize_seed],
         outputs=[rgb, depth, generated_seed, html],
         fn=predict,
         cache_examples=True)
 
     new_btn.click(
         fn=predict,
-        inputs=[prompt, negative_prompt, guidance_scale, seed, randomize_seed],
+        inputs=[prompt, negative_prompt, guidance_scale, ddim_steps, seed, randomize_seed],
         outputs=[rgb, depth, generated_seed, html],
     )