prepare wiggle output

app.py

@@ -7,7 +7,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 from PIL import Image
 
-def update(slider, model_type, img):
+def calculate_depth(model_type, img):
 
     if not os.path.exists('temp'):
       os.system('mkdir temp')
@@ -53,11 +53,15 @@ def update(slider, model_type, img):
 
     return f'temp/image_depth.jpeg'
 
+def wiggle_effect(slider):
+
+    return f'temp/image_depth.jpeg'
+
 
 
 with gr.Blocks() as demo:
     gr.Markdown("Start typing below and then click **Run** to see the output.")
-    inp = [gr.Slider(1,15, default = 2, label='StepCycles',step= 1)]
+    inp = []
 
     midas_models = ["DPT_Large","DPT_Hybrid","MiDaS_small"]
 
@@ -66,7 +70,13 @@ with gr.Blocks() as demo:
     with gr.Row():
         inp.append(gr.Image(type="pil", label="Input"))
         out = gr.Image(type="file", label="Output")
-    btn = gr.Button("Run")
-    btn.click(fn=update, inputs=inp, outputs=out)
+    btn = gr.Button("Calculate depth")
+    btn.click(fn=calculate_depth, inputs=inp, outputs=out)
+
+    inp = [gr.Slider(1,15, default = 2, label='StepCycles',step= 1)]
+    with gr.Row():
+        out = gr.Image(type="file", label="Output")
+    btn = gr.Button("Calculate depth")
+    btn.click(fn=wiggle_effect, inputs=inp, outputs=out)
 
 demo.launch()