srochaeduardo commited on
Commit
5c59d4c
·
verified ·
1 Parent(s): 259fbba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -3,13 +3,11 @@ import fitz # PyMuPDF
3
  # import requests # Não precisamos mais do requests
4
  from transformers import pipeline
5
 
6
- # Carrega o modelo de resumo (você pode usar online ou localmente)
7
- # Para usar online:
8
- summarizer = pipeline("summarization", model="unicamp-dl/mt5-base-multi-pt-br-summary")
9
 
10
- # Para usar localmente (RECOMENDADO - após baixar os arquivos do modelo):
11
- # summarizer = pipeline("summarization", model="./mt5-base-multi-pt-br-summary")
12
- # (Certifique-se de que a pasta 'mt5-base-multi-pt-br-summary' esteja no mesmo diretório do app.py)
13
 
14
 
15
  def extract_text_from_pdf(pdf_path):
 
3
  # import requests # Não precisamos mais do requests
4
  from transformers import pipeline
5
 
6
+ # Carrega o modelo de resumo
7
+ summarizer = pipeline("summarization", model="pierreguillou/t5-base-portuguese-summarization")
 
8
 
9
+ # Para usar um modelo local (após baixar):
10
+ # summarizer = pipeline("summarization", model="./pasta_do_modelo")
 
11
 
12
 
13
  def extract_text_from_pdf(pdf_path):