added other gans

app.py

@@ -53,14 +53,16 @@ def calculate_depth(model_type, img):
 
     return out_im
 
-def wiggle_effect(slider):
+def wiggle_effect(slider, gan_type):
 
     dim = '256' 
     c_images = '1'
     name_output = 'out'
 
+    dict_saved_gans = {'Cycle': '74962_110', 'Cycle(half)': '66942','noCycle': '31219_110', 'noCycle-noCr': '92332_110', 'noCycle-noCr-noL1': '82122_110', 'OnlyGen': '70944_110' }
+
     subprocess.run(["python", "main.py", "--gan_type", 'WiggleGAN', "--expandGen", "4", "--expandDis", "4", "--batch_size", c_images, "--cIm", c_images,
-                    "--visdom", "false", "--wiggleDepth", str(slider), "--seedLoad", '31219_110', "--gpu_mode", "false", "--imageDim", dim, "--name_wiggle", name_output
+                    "--visdom", "false", "--wiggleDepth", str(slider), "--seedLoad", dict_saved_gans[gan_type], "--gpu_mode", "false", "--imageDim", dim, "--name_wiggle", name_output
                     ])
     subprocess.run(["python",  "WiggleResults/split.py", "--dim", dim])
 
@@ -82,7 +84,9 @@ with gr.Blocks() as demo:
 
 
     ## Wigglegram
+    gan_models = ["Cycle","Cycle(half)","noCycle","noCycle-noCr","noCycle-noCr-noL1","OnlyGen"]
     inp = [gr.Slider(1,15, default = 2, label='StepCycles',step= 1)]
+    inp.append(gr.inputs.Dropdown(gan_models, default="Cycle", label="Different Gan trainings"))
     with gr.Row():
         out = [ gr.Image(type="file", label="Output_images"), #TODO change to gallery 
                 gr.Image(type="file", label="Output_wiggle")]