Update app.py

app.py

@@ -450,6 +450,8 @@ def get_prompts(prompt_text):
     return gen   
 
 def gen_im(prompt,model_choice=5,count=1):
+    if count >4:
+        count =4
     t=0
     output_list=[]
     while t<int(count):