Update app.py

app.py

@@ -1,11 +1,10 @@
 import streamlit as st
 import pandas as pd
 import torch
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Carregando o modelo e o tokenizador do GPT-2
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model = GPT2LMHeadModel.from_pretrained('gpt2')
+tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
+model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
 
 # Set the padding token to the end-of-sequence token
 if tokenizer.pad_token is None: