Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import BitsAndBytesConfig, PaliGemmaForConditionalGeneration,
 import spaces
 import torch
 import os
-import re
+
 
 access_token = os.getenv('HF_token')
 
@@ -15,22 +15,22 @@ bnb_config = BitsAndBytesConfig(load_in_8bit=True)
 
 
 def convert_to_markdown(input_text):
-  input_text = input_text.replace("!:", ":")
-
-  # Find all words before ': **' and replace with bold markdown
-  output_text = re.sub(r'(\w+)\s*:\s*\*\*', r'**\1**:', input_text)
-
-  # Replace double asterisks with double hashtags for remaining headings
-  output_text = output_text.replace("**", "##")
-
-  # Remove any extra whitespace at the beginning of lines
-  output_text = re.sub(r'^\s+', '', output_text, flags=re.MULTILINE)
-
-  # Add an extra newline after each heading (after the colon)
-  output_text = "## " + re.sub(r'(##\s*.*):', r'\1:\n\n', output_text)
-
-  return output_text
-
+    # Split the input text into sections based on the '**' delimiter
+    sections = input_text.split("**")
+    
+    # Initialize the formatted output with the bird name
+    formatted_output = f"**{sections[0].strip()}**\n"
+    
+    # Process each section to format it
+    for i in range(1, len(sections), 2):
+        if i + 1 < len(sections):
+            # Use '##' for subheadings and clean up the text
+            header = sections[i].strip() + "** "
+            content = sections[i + 1].strip()
+            formatted_output += f"\n**{header}{content}\n"
+    
+    # Return the formatted output
+    return formatted_output.strip()
 
 @spaces.GPU
 def infer_fin_pali(image, question):