Update app.py

app.py

@@ -35,12 +35,14 @@
 
 
 import streamlit as st
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 # Load the tokenizer and model
-tokenizer = GPT2Tokenizer.from_pretrained("gpt2-large")
-model = GPT2LMHeadModel.from_pretrained("gpt2-large")
+# tokenizer = GPT2Tokenizer.from_pretrained("gpt2-large")
+# model = GPT2LMHeadModel.from_pretrained("gpt2-large")
+tokenizer = AutoTokenizer.from_pretrained("gpt2-large")
+model = AutoModelForCausalLM.from_pretrained("gpt2-large")
 
 def generate_blog(title):
     # Encode the input text