Update app.py

app.py

@@ -49,13 +49,12 @@ def find_most_similar_matn(text, n):
     cos_sim = cosine_similarity(embed_text.reshape(1, -1), arr)
     indices = np.argsort(cos_sim)[0][-n:]
     matns = df.iloc[indices]
-    matns['Cosine Similarity'] = cos_sim[0][indices]
+    matns['Similarity'] = cos_sim[0][indices]
     matns_prep = [araby.strip_diacritics(text) for text in matns['matn']]
     to_compare = [(i, prep_text) for i in matns_prep]
     is_taraf = model_CE.predict(to_compare)
-    matns['Cross Similarity'] = is_taraf
     matns = matns[is_taraf> .5]
-    return matns[['Book_Name', 'matn', 'taraf_ID', 'Book_ID', 'Hadith Number', 'Author', 'Cosine Similarity', 'Cross Similarity']]
+    return matns[['Book_Name', 'matn', 'taraf_ID', 'Book_ID', 'Hadith Number', 'Author', 'Similarity']]
 
 with gr.Blocks() as demo:
     text_input = gr.Textbox()