Update app.py

app.py

@@ -3,6 +3,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from fpdf import FPDF
 import torch
 import spaces
+import io
+import markdown
 
 # Initialize the Qwen model and tokenizer
 model_name = "Qwen/Qwen2.5-Coder-7B-Instruct"
@@ -28,7 +30,7 @@ def generate_documentation(code_input):
     generated_ids = [output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)]
     documentation = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
     print(documentation)
-    return documentation
+    return markdown.markdown(documentation)
 
 # Function to generate and download PDF
 def create_pdf(documentation):