Update app.py

app.py

@@ -451,10 +451,15 @@ def get_prompts(prompt_text):
     return gen   
 
 def gen_im(prompt,model_choice=5,count=1):
-    out1 = models2(prompt,model_choices,count)
-
-    return out1
-    
+    t=0
+    output_list=[]
+    while t<int(count):
+        proc1=models2[int(model_choice)]
+        output1=proc1(inputs)
+        output_list.append(output1)
+        inputs=f"{inputs} "
+        t+=1
+    return(output_list)
     
 with gr.Blocks() as build_app:
     gr.Markdown("""<center><h1>Interactive Twitter Card Builder</h1></center>""")