Update app.py

app.py

@@ -9,7 +9,7 @@ import chromadb
 from datasets import load_dataset
 # from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
-from transformers import AutoTokenizer, MistralForCausalLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 
 
@@ -87,8 +87,8 @@ vector_store.populate_vectors(dataset=None)
 # tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
 # model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
 
-model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
+tokenizer = AutoTokenizer.from_pretrained("merve/chatgpt-prompts-bart-long")
+model = AutoModelForSeq2SeqLM.from_pretrained("merve/chatgpt-prompts-bart-long", from_tf=True)
 
 # Define the chatbot response function
 def chatbot_response(user_input):