Update app.py

app.py

@@ -5,26 +5,15 @@ import pickle
 from groq import Groq
 from datasets import load_dataset
 from transformers import pipeline
-from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Initialize Groq API
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-model_name = "rajkumarrrk/dialogpt-fine-tuned-on-daily-dialog"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
-
-chat_pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
-
-
 # Load datasets
 healthcare_ds = load_dataset("harishnair04/mtsamples")
 education_ds = load_dataset("ehovy/race", "all")
 finance_ds = load_dataset("warwickai/financial_phrasebank_mirror")
 
-# Load chat model
-chat_pipe = pipeline("text-generation", model="rajkumarrrk/dialogpt-fine-tuned-on-daily-dialog")
-
 # FAISS Index Setup
 index = faiss.IndexFlatL2(768)
 chat_history = []