Update app.py

app.py

@@ -48,6 +48,7 @@ def load_fn(models):
                 m = gr.Interface(lambda txt: None, ['text'], ['image'])
                 print(error)
             models_load.update({model: m})
+    print(models_load)
 
 
 load_fn(models)
@@ -282,14 +283,17 @@ def make_me():
                             cfg = gr.Slider(label="Guidance scale", info="If 0, the default value is used.", maximum=30.0, step=0.1, value=0, interactive = True)
                             seed = gr.Slider(label="Seed", info="Randomize Seed if -1.", minimum=-1, maximum=MAX_SEED, step=1, value=-1, interactive = True)
                             
-            list_param=gr.Dropdown(choices=[["a",[["","",0,0,0,0,-1]]]], value=[["","",0,0,0,0,-1]], visible=False)
+            add_param=gr.Button("Add to the list")
+            del_param=gr.Button("Delete to the list")
             
-            add_param=gr.Button("add to the list")
-            del_param=gr.Button("delete to the list")
+            with gr.Row():
+                nb_images_by_prompt=gr.Number("Number of images by prompt:",value=2,interactive=True)
+                nb_of_models_to_gen=gr.Number("Number of Models:",value=10,interactive=True)
         #gen_button = gr.Button('Generate images', scale=3)
         #stop_button = gr.Button('Stop', variant='secondary', interactive=False, scale=1)
         #gen_button.click(lambda: gr.update(interactive=True), None, stop_button)
-
+        
+        list_param=gr.Dropdown(choices=[["a",[["","",0,0,0,0,-1]]]], value=[["","",0,0,0,0,-1]], visible=False)
         disp_param = gr.Examples(
             label="list of prompt",
             examples=list_param.value,