hergo-7 commited on
Commit
9694cf1
·
verified ·
1 Parent(s): 6e93a7c

Upload pdf_utils.py

Browse files
Files changed (1) hide show
  1. 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="sshleifer/distilbart-cnn-12-6")
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 = 1500
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