Update app.py

app.py

@@ -4,8 +4,7 @@ import fitz  # PyMuPDF
 from transformers import pipeline
 
 # Carrega o modelo de resumo (você vai inserir o nome correto aqui)
-#summarizer = pipeline("summarization", model="stjiris/t5-portuguese-legal-summarization")
-summarizer = pipeline("summarization", model="rhaymison/flan-t5-portuguese-small-summarization")
+summarizer = pipeline("summarization", model="stjiris/t5-portuguese-legal-summarization")
 
 # Para usar um modelo local (após baixar):
 # summarizer = pipeline("summarization", model="./pasta_do_modelo")
@@ -32,7 +31,7 @@ def summarize_pdf(pdf_file):  # Recebe o objeto gr.File
         text = extract_text_from_pdf(pdf_file.name)  # Passa o NOME do arquivo temporário
 
         # Resume o texto. Ajuste max_length e min_length conforme necessário.
-        summary = summarizer(text, max_length=512, min_length=50, do_sample=False)[0]["summary_text"]
+        summary = summarizer(text, max_length=1028, min_length=50, do_sample=False)[0]["summary_text"]
         return summary
     except Exception as e:
         return f"Erro ao processar o PDF: {e}"