Update app.py

app.py

@@ -16,14 +16,14 @@ hf_token = os.getenv("HF_TOKEN")
 
 login(token=hf_token)
 
-def load_faiss_and_embeddings_from_hf(repo_id, index_file, embedding_file):
-    faiss_index_path = hf_hub_download(repo_id=repo_id, filename=index_file)
-    embeddings_path = hf_hub_download(repo_id=repo_id, filename=embedding_file)
-    
+def load_embeddings_and_faiss():
+    embeddings_path = hf_hub_download(repo_id="chukbert/embedding-faq-medquad", filename="data/embeddings-00000-of-00001.parquet")
+    faiss_index_path = hf_hub_download(repo_id="chukbert/embedding-faq-medquad", filename="data/faiss_index-00000-of-00001.parquet")
+
+    embeddings_df = pd.read_parquet(embeddings_path)
+    question_embeddings = embeddings_df.values
+
     faiss_index = faiss.read_index(faiss_index_path)
-    
-    with open(embeddings_path, 'rb') as f:
-        question_embeddings = pickle.load(f)
 
     return faiss_index, question_embeddings
 
@@ -87,7 +87,6 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
 
 
 
-# Gradio ChatInterface with additional inputs for model settings and response time
 demo = gr.ChatInterface(
     fn=respond,
     additional_inputs=[