Update app.py

app.py

@@ -9,13 +9,11 @@ from datasets import load_dataset
 import gradio as gr
 from transformers import GPT2Tokenizer, GPT2LMHeadModel, Trainer, TrainingArguments
 
-model_name = "Amitesh007/text_generation-finetuned-gpt2"
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Load the tokenizer
-tokenizer = GPT2Tokenizer.from_pretrained(model_name)
-
-# Load the model with from_tf=True
-model = GPT2LMHeadModel.from_pretrained(model_name, from_tf=True)
+tokenizer = AutoTokenizer.from_pretrained("umashankar00/Text_Generation_model")
+model = AutoModelForCausalLM.from_pretrained("umashankar00/Text_Generation_model")
 
 # Function to clear the cache
 def clear_cache(model_name):