Update app.py

app.py

@@ -426,7 +426,7 @@ def get_prompts(prompt_text):
     return gen   
 
 def gen_im(prompt,model_choice,count=1):
-    out = im_gen(prompt,model_choice,count)
+    out = im_gen(prompt,int(model_choice),count)
     return out
     
     
@@ -472,7 +472,10 @@ with gr.Blocks() as build_app:
                     #gen_im = gr.Button("Generate Image")
                 with gr.Accordion("Generated Image Controls", open=False):
                     prompt_input=gr.Textbox("Seed Prompt", placeholder="Optional")
-                    model_name1 = gr.Dropdown(show_label=False, choices=[m for m in models], type="index", value=current_model, interactive=True)
+                    with gr.Row():
+                        model_name1 = gr.Dropdown(show_label=False, choices=[m for m in models], type="index", value=current_model, interactive=True)
+                        count = gr.Number(value=1)
+
                     with gr.Row():
                         magic_btn=gr.Button("Magic Prompt")
                         gen_btn=gr.Button("Generate Images")
@@ -489,7 +492,7 @@ with gr.Blocks() as build_app:
         gr.Column(scale=1)
 
     magic_btn.click(get_prompts,prompt_input,prompt_input)
-    gen_btn.click(gen_im,[prompt_input,model_name1],gen_im_out)
+    gen_btn.click(gen_im,[prompt_input,model_name1,count],gen_im_out)
     
     t_image.change(up_c,[t_image,c_slider],c_image)
     c_upbtn.click(slide_fn,c_slider,c_image)