Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-from transformers import BartTokenizer, BartForConditionalGeneration
+from transformers import BartTokenizer, BartForConditionalGeneration, AutoConfig
 
 # Obtendo o token do ambiente
 HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -9,8 +9,9 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 model_name = 'rubendsm/inteligencia-ricardo-bart-large-2'
 
 # Carregando o tokenizador e o modelo
+config = AutoConfig.from_pretrained(model_name, hf_token=HF_TOKEN)
 tokenizer = BartTokenizer.from_pretrained(model_name)
-model = BartForConditionalGeneration.from_pretrained(model_name, hf_token=HF_TOKEN)
+model = BartForConditionalGeneration.from_pretrained(model_name, config=config)
 
 # Função para sumarizar o texto de entrada
 def summarize(input_text):