Update app.py

app.py

@@ -9,10 +9,16 @@ import chromadb
 from datasets import load_dataset
 # from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import GPT2Tokenizer, GPT2Model
+
+model_name = "Amitesh007/text_generation-finetuned-gpt2"
+
+# Load the tokenizer
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+
+# Load the model with from_tf=True
+model = GPT2Model.from_pretrained(model_name, from_tf=True)
 
-tokenizer = AutoTokenizer.from_pretrained("Amitesh007/text_generation-finetuned-gpt2")
-model = AutoModelForCausalLM.from_pretrained("Amitesh007/text_generation-finetuned-gpt2")
 
 # Function to clear the cache
 def clear_cache(model_name):