Spaces:

codeteach
/

textevaluation

Runtime error

codeteach commited on May 30

Commit

df4431b

•

1 Parent(s): 16d8bf6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from transformers import pipeline
 import fitz  # PyMuPDF
 # Load a summarization model from Hugging Face
-summarizer = pipeline("summarization")
 def extract_text_from_pdf(pdf_file):
     text = ""
@@ -108,3 +108,4 @@ interface.launch()

 import fitz  # PyMuPDF
 # Load a summarization model from Hugging Face
+summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 def extract_text_from_pdf(pdf_file):
     text = ""