Add padding between images

app.py

@@ -124,21 +124,20 @@ def generate_for_prompt(input_text, state, ret_scale_factor, max_num_rets, num_w
     im_names = []
     response = ''
     text_outputs = []
-    for output in model_outputs:
+    for output_i, output in enumerate(model_outputs):
         if type(output) == str:
+            if output_i > 0:
+                response += '<br/>'
             text_outputs.append(output)
-            response += output
+            if len(model_outputs) > 1:
+                response += output + '<br/>'
         elif type(output) == list:
-            response += '<br/>'  # Add line break between images.
             for image in output:
                 filename = save_image_to_local(image)
                 response += f'<img src="/file={filename}" style="display: inline-block;">'
-            response += '<br/>'
         elif type(output) == Image.Image:
             filename = save_image_to_local(output)
-            response += '<br/>'
             response += f'<img src="/file={filename}" style="display: inline-block;">'
-            response += '<br/>'
 
     # TODO(jykoh): Persist image inputs.
     chat_history = model_inputs + \
@@ -188,7 +187,7 @@ with gr.Blocks(css=css) as demo:
             ret_scale_factor = gr.Slider(minimum=0.0, maximum=3.0, value=1.0, step=0.1, interactive=True,
                                          label="Frequency multiplier for returning images (higher means more frequent)")
             max_ret_images = gr.Number(
-                minimum=0, maximum=3, value=1, precision=1, interactive=True, label="Max images to return")
+                minimum=0, maximum=3, value=2, precision=1, interactive=True, label="Max images to return")
             gr_max_len = gr.Slider(minimum=1, maximum=64, value=32,
                                    step=1, interactive=True, label="Max # of words returned")
             gr_temperature = gr.Slider(