Update app.py

app.py

@@ -5,8 +5,12 @@ import os
 import pandas as pd
 from datasets import load_dataset
 from sklearn.metrics.pairwise import cosine_similarity
-
+from datasets import Features, Value
 import plotly.express as px
+
+features = Features({'matn': Value('string'), 'taraf_ID': Value('string'), 'bookid_hadithid': Value('string')})
+
+
 Secret_token = os.getenv('HF_token')
 
 dataset = load_dataset("FDSRashid/embed_matn", token = Secret_token)
@@ -40,7 +44,7 @@ def plot_similarity_score(taraf_num):
     lower_triangle = matr[mask]
     data = lower_triangle.flatten()
     fig_dis = px.histogram(x = data,  title = f'Similarity Distribution for Taraf {taraf_num}', labels = {'x': 'Similarity Score'}, nbins = 20, template = 'ggplot2' )
-    return fig, fig_dis, taraf_df[['matn', 'Number', 'Book', 'Author', 'Hadith Number']]
+    return fig, fig_dis, taraf_df[['matn', 'Number', 'Book_Name', 'Author', 'Hadith Number']]
 
 with gr.Blocks() as demo: 
   taraf_number = gr.Slider(1,taraf_max , value=10000, label="Taraf", info="Choose the Taraf to Input", step = 1)