Adjust formatting

.gitignore

@@ -1 +1,2 @@
-.env
+.env
+/venv

app.py

@@ -71,30 +71,36 @@ def captioner(image):
     base64_image = image_to_base64_str(image)
     results = get_completion(base64_image)
     captions = []
-    separator = "-" * 50
     for endpoint, result in results.items():
+        # Use a smaller heading or remove the heading syntax for regular text size
+        # header = f"#### [{endpoint}]({hf_base_url+endpoint}):"
+        header = f"[{endpoint}]({hf_base_url+endpoint}):"  # No heading, regular text
         if "error" not in result:
-            caption = f"[{endpoint}]({hf_base_url+endpoint}): \n\n {result[0]['generated_text']} \n {separator} \n\n "
+            caption = result[0]["generated_text"]
         else:
-            caption = f"[{endpoint}]({hf_base_url+endpoint}): \n\n Error - {result['error']} \n {separator} \n\n "
-        captions.append(caption)
-    return "".join(captions)  # Join all captions into a single string
+            caption = f"Error - {result['error']}"
+        captions.append(
+            f"{header}\n{caption} \n\n"
+        )  # Use horizontal rule for separation
+    return "\n".join(
+        captions
+    ).strip()  # Join all captions into a single string, separated by horizontal rules
 
 
 # --- Launch the Gradio App ---
 demo = gr.Interface(
     fn=captioner,
     inputs=[gr.Image(label="Upload image", type="pil")],
-    outputs=gr.Markdown(label="Captions"),
-    title="Captioning Comparison",
+    outputs=gr.Markdown(label="Captions"),  # Use a single Markdown output
+    title="Image Captioning Model Comparison",
     description="Upload an image and see how different models describe it!",
     allow_flagging="never",
     examples=[
-    "example_1.jpg",
-    "example_2.jpg",
-    "example_3.jpg",
-    "example_4.png",
-    "example_5.png",
+        "example_1.jpg",
+        "example_2.jpg",
+        "example_3.jpg",
+        "example_4.png",
+        "example_5.png",
     ],
 )