Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from sentence_transformers import SentenceTransformer
 import pandas as pd
+import pickle
 
 def find(query):
     def get_detailed_instruct(task_description: str, query: str) -> str:
@@ -13,7 +14,7 @@ def find(query):
     ]
     
     quran = pd.read_csv('quran-simple-clean.txt', delimiter="|")
-    file = open('encoded_quran_text_multilingual-e5-large-instruct.sav','rb')
+    file = open('encoded_quran_text_split_multilingual-e5-large-instruct.sav','rb')
     document_embeddings = pickle.load(file)
     
     model = SentenceTransformer('intfloat/multilingual-e5-large-instruct')
@@ -21,8 +22,10 @@ def find(query):
     scores = (query_embeddings @ document_embeddings.T) * 100
 
     # insert the similarity value to dataframe & sort it
-    quran['similarity'] = scores.tolist()[0]
-    sorted_quran = quran.sort_values(by='similarity', ascending=False)
+    file = open('quran-splitted.sav','rb')
+    quran_splitted = pickle.load(file)
+    quran_splitted['similarity'] = scores.tolist()[0]
+    sorted_quran = quran_splitted.sort_values(by='similarity', ascending=False)
     
     results = ""
     i = 0