Update app.py

app.py

@@ -2,10 +2,11 @@
 import gradio as gr
 import os
 from langchain_community.vectorstores import FAISS
-from langchain_together import TogetherEmbeddings
-from operator import itemgetter
+from langchain_together import TogetherEmbeddings, Together
+from langchain.prompts import ChatPromptTemplate
+from langchain.schema.runnable import RunnablePassthrough
+from langchain.schema.output_parser import StrOutputParser
 from langchain.memory import ConversationBufferMemory
-from langchain.schema import format_document
 from typing import List, Tuple
 
 # Environment variables for API keys
@@ -22,7 +23,8 @@ class ChatBot:
         # Load the pre-created FAISS index with embeddings
         self.vectorstore = FAISS.load_local(
             "faiss_index",
-            embeddings=self.embeddings
+            embeddings=self.embeddings,
+            allow_dangerous_deserialization=True  # Only enable this if you trust the source of the index
         )
         self.retriever = self.vectorstore.as_retriever()