Update app.py

app.py

@@ -4,10 +4,30 @@ from sentence_transformers import SentenceTransformer
 
 def greet(name):
     # transform query from user
-    file = open('ms-marco-TinyBERT-L-2.sav','rb')
-    model = pickle.load(file)
+    model = SentenceTransformer('use-cmlm-multilingual')
+    encoded_query_text = model.encode(name)
+
+    # get encoded quran text
+    file = open('encoded_quran_cmlm.sav','rb')
+    encoded_quran_text = pickle.load(file)
     file.close()
-    return "Hello " + name + "!"
+
+    # compare query to each quran verse
+    i = 0
+    text_similarity = []
+    for encoded_quran_ayat in encoded_quran_text:
+      similarity = encoded_query_text @ encoded_quran_ayat.T
+      text_similarity.append(similarity)
+      i=i+1
+      print(i)
+
+    # insert the similarity value to dataframe & sort it
+    quran = pd.read_csv('quran-simple-clean.txt', delimiter="|")
+    quran['similarity'] = text_similarity
+    sorted_quran = quran.sort_values(by='similarity', ascending=False)
+    df_list = sorted_quran.values.tolist()
+
+    return df_list['text'][0]
 
 demo = gr.Interface(fn=greet, inputs="textbox", outputs="textbox")