Update app.py

app.py

@@ -2,26 +2,18 @@ import gradio as gr
 import fitz
 import torch
 from transformers import pipeline
-import time, io
+import time
 
 device = 0 if torch.cuda.is_available() else -1
-if device == -1: print("⚠️ No GPU detected. Expect ~10–20s for 300,000 chars on CPU.")
+if device == -1: print("⚠️ No GPU. Expect ~12–22s for 300,000 chars on CPU.")
 
-summarizer = pipeline("summarization", model="google/pegasus-xsum", device=device, torch_dtype=torch.float32)
+summarizer = pipeline("summarization", model="t5-small", device=device, torch_dtype=torch.float32)
 
-def extract_text(file_bytes):
-    if file_bytes[:4].startswith(b'%PDF'):
-        doc = fitz.open(stream=file_bytes, filetype="pdf")
-        text = "".join(page.get_text("text", flags=16) for page in doc)
-        doc.close()
-        return text
-    try: return file_bytes.decode("utf-8")
-    except: return "❌ Unsupported format (PDF/TXT only)"
-
-async def summarize_file(file_bytes):
+def summarize_file(file_bytes):
     start = time.time()
-    text = extract_text(file_bytes)[:300000] or "❌ No text found"
-    if len(text.strip()) == 0: return text
+    text = "".join(page.get_text("text", flags=16) for page in fitz.open(stream=file_bytes, filetype="pdf")) if file_bytes[:4].startswith(b'%PDF') else file_bytes.decode("utf-8", errors="ignore")
+    if not text.strip(): return "❌ No text found"
+    text = text[:300000]
     chunks = [text[i:i+15000] for i in range(0, len(text), 15000)]
     if not chunks: return "❌ No chunks to summarize"
     summaries = []
@@ -30,8 +22,8 @@ async def summarize_file(file_bytes):
         if time.time() - start > 9: 
             summaries.append("⚠️ Stopped early")
             break
-        batch = chunks[i:i+batch_size]
         try:
+            batch = chunks[i:i+batch_size]
             batch_summaries = summarizer(batch, max_length=40, min_length=10, do_sample=False, batch_size=batch_size)
             summaries.extend(f"**Chunk {i+j+1}**:\n{s['summary_text']}" for j, s in enumerate(batch_summaries))
         except: summaries.append(f"**Chunk {i+1}**: ❌ Error")
@@ -40,7 +32,7 @@ async def summarize_file(file_bytes):
 demo = gr.Interface(
     fn=summarize_file, inputs=gr.File(label="📄 PDF/TXT Notes"), 
     outputs=gr.Textbox(label="📝 Summary"), 
-    title="Fast Summarizer", description="300,000+ chars in ~5–10s (GPU) or ~10–20s (CPU)"
+    title="Fast Summarizer", description="300,000+ chars in ~5–8s (GPU) or ~12–22s (CPU)"
 )
 
 if __name__ == "__main__":