Update app.py

app.py

@@ -9,10 +9,10 @@ import chromadb
 from datasets import load_dataset
 # from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-tokenizer = AutoTokenizer.from_pretrained("merve/chatgpt-prompt-generator-v12")
-model = AutoModelForSeq2SeqLM.from_pretrained("merve/chatgpt-prompt-generator-v12")
+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):