added multiple dimensions

app.py

@@ -8,7 +8,7 @@ import matplotlib.pyplot as plt
 from PIL import Image
 import subprocess
 
-def calculate_depth(model_type, gan_type, slider, img):
+def calculate_depth(model_type, gan_type, dim, slider, img):
 
     if not os.path.exists('temp'):
       os.system('mkdir temp')
@@ -51,7 +51,6 @@ def calculate_depth(model_type, gan_type, slider, img):
     out_im = Image.fromarray(formatted)
     out_im.save("Images/Input-Test/1_d.png", "PNG")
 
-    dim = '256' 
     c_images = '1'
     name_output = 'out'
 
@@ -71,9 +70,12 @@ with gr.Blocks() as demo:
     ## Depth Estimation
     midas_models = ["DPT_Large","DPT_Hybrid","MiDaS_small"]
     gan_models = ["Cycle","Cycle(half)","noCycle","noCycle-noCr","noCycle-noCr-noL1","OnlyGen"]
+    dim = ['256','512','1024']
 
-    inp = [gr.inputs.Dropdown(midas_models, default="MiDaS_small", label="Depth estimation model type")]
-    inp.append(gr.inputs.Dropdown(gan_models, default="Cycle", label="Different Gan trainings"))
+    with gr.Row():
+        inp = [gr.inputs.Dropdown(midas_models, default="MiDaS_small", label="Depth estimation model type")]
+        inp.append(gr.inputs.Dropdown(gan_models, default="Cycle", label="Different GAN trainings"))
+        inp.append(gr.inputs.Dropdown(dim, default="256", label="Wiggle dimension result"))
     inp.append(gr.Slider(1,15, default = 2, label='StepCycles',step= 1))
     with gr.Row():
         inp.append(gr.Image(type="pil", label="Input"))
@@ -81,7 +83,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         out.append(gr.Image(type="file", label="Output_images"))
         out.append(gr.Image(type="file", label="Output_wiggle"))
-    btn = gr.Button("Calculate depth + wiggle")
+    btn = gr.Button("Calculate depth + Wiggle")
     btn.click(fn=calculate_depth, inputs=inp, outputs=out)