Update app.py

app.py

@@ -17,9 +17,14 @@ def summarize_pdf(pdf_file):
 
     # Summarize the text
     inputs = tokenizer.encode("summarize: " + text, return_tensors="pt", max_length=1024, truncation=True)
-    summary_ids = model.generate(inputs, max_length=150, min_length=40, length_penalty=2.0, num_beams=4, early_stopping=True)
+    summary_ids = model.generate(inputs, max_length=300, min_length=80, length_penalty=2.0, num_beams=4, early_stopping=True)
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
-    return summary
+    
+    # Convert the summary to bullet points
+    bullet_points = summary.split('. ')
+    bullet_summary = "\n".join([f"• {point}" for point in bullet_points if point])
+    
+    return bullet_summary
 
 # Interface for Gradio
 iface = gr.Interface(
@@ -27,7 +32,8 @@ iface = gr.Interface(
     inputs="file",
     outputs="text",
     title="PDF Summarizer",
-    description="Upload a PDF file, and the model will provide a summary.",
+    description="Upload a PDF file and get a bullet-point summary of its content.",
+    theme="miku",
     live=True,
     capture_session=True,
 )