Spaces:
Running
Running
Upload pdf_utils.py
Browse files- backend/pdf_utils.py +2 -2
backend/pdf_utils.py
CHANGED
|
@@ -666,7 +666,7 @@ def summarize_pdf(pdf_bytes, length="medium"):
|
|
| 666 |
if summarizer is None:
|
| 667 |
try:
|
| 668 |
from transformers import pipeline
|
| 669 |
-
summarizer = pipeline("summarization", model="
|
| 670 |
except ImportError:
|
| 671 |
return "<p style='color:red'>Erro no Servidor: As bibliotecas de IA ('transformers', 'torch') não foram instaladas. Atualize o requirements.txt e reinicie o servidor (Hugging Face).</p>"
|
| 672 |
|
|
@@ -676,7 +676,7 @@ def summarize_pdf(pdf_bytes, length="medium"):
|
|
| 676 |
text += page.get_text("text") + " "
|
| 677 |
doc.close()
|
| 678 |
|
| 679 |
-
max_chunk =
|
| 680 |
chunks = [text[i:i+max_chunk] for i in range(0, len(text), max_chunk)]
|
| 681 |
|
| 682 |
max_l = 130
|
|
|
|
| 666 |
if summarizer is None:
|
| 667 |
try:
|
| 668 |
from transformers import pipeline
|
| 669 |
+
summarizer = pipeline("summarization", model="rhaymison/flan-t5-portuguese-small-summarization")
|
| 670 |
except ImportError:
|
| 671 |
return "<p style='color:red'>Erro no Servidor: As bibliotecas de IA ('transformers', 'torch') não foram instaladas. Atualize o requirements.txt e reinicie o servidor (Hugging Face).</p>"
|
| 672 |
|
|
|
|
| 676 |
text += page.get_text("text") + " "
|
| 677 |
doc.close()
|
| 678 |
|
| 679 |
+
max_chunk = 1200
|
| 680 |
chunks = [text[i:i+max_chunk] for i in range(0, len(text), max_chunk)]
|
| 681 |
|
| 682 |
max_l = 130
|