Update app.py

app.py

@@ -9,7 +9,7 @@ def find(query):
     encoded_query_text = model.encode(query)
 
     # get encoded quran text
-    file = open('encoded_quran_cmlm_finetuned.sav','rb')
+    file = open('encoded_quran_fine-tuning-use-cmlm-multilingual-quran-splitted.sav','rb')
     encoded_quran_text = pickle.load(file)
     file.close()
 
@@ -23,14 +23,20 @@ def find(query):
       print(i)
 
     # insert the similarity value to dataframe & sort it
-    quran = pd.read_csv('quran-simple-clean.txt', delimiter="|")
-    quran['similarity'] = text_similarity
+    file = open('quran-splitted.sav','rb')
+    quran_splitted = pickle.load(file)
+    quran_splitted['similarity'] = text_similarity
     sorted_quran = quran.sort_values(by='similarity', ascending=False)
     
+    # insert the similarity value to dataframe & sort it
+    quran = pd.read_csv('quran-simple-clean.txt', delimiter="|")
+    
     results = ""
     i = 0
     while i<6:
-        results = results + sorted_quran.iloc[i]['text']+" (Q.S "+str(sorted_quran.iloc[i]['sura'])+":"+str(sorted_quran.iloc[i]['aya'])+")\n"
+        result = sorted_quran.iloc[i]
+        result_quran = quran.loc[(quran['sura']==result['sura']) & (teks['aya']==result['aya'])]
+        results = results + result_quran['text']+" (Q.S "+str(result['sura'])+":"+str(result['aya'])+")\n"
         i=i+1
     
     return results