Update app.py

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
 import torch
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import os
 
 # Get the Hugging Face token from the environment variable
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
 # Load the tokenizer and model
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2', use_auth_token=HF_TOKEN)
-model = GPT2LMHeadModel.from_pretrained('skylersterling/SentimentGPT', use_auth_token=HF_TOKEN)
+tokenizer = AutoTokenizer.from_pretrained('EleutherAI/GPT-NeoX-20B', use_auth_token=HF_TOKEN)
+model = AutoModelForCausalLM.from_pretrained('skylersterling/SentimentGPT', use_auth_token=HF_TOKEN)
 model.eval()
 model.to('cpu')