Update app.py

app.py

@@ -19,13 +19,15 @@ dataset2 = load_dataset('FDSRashid/hadith_info',data_files = 'Taraf_Info.csv', t
 
 
 
-features = Features({'Rawi ID': Value('int32'), 'Famous Name': Value('string'), 'Narrator Rank': Value('string'), 'Number of Narrations': Value('string'), 'Official Name':Value('string'), 'Title Name':Value('string')} )
+features = Features({'Rawi ID': Value('int32'), 'Famous Name': Value('string'), 'Narrator Rank': Value('string'), 'Number of Narrations': Value('string'), 'Official Name':Value('string'), 'Title Name':Value('string'), 'Generation': Value('string')} )
 narrator_bios = load_dataset("FDSRashid/hadith_info", data_files = 'Teacher_Bios.csv', token = Secret_token,features=features )
 narrator_bios = narrator_bios['train'].to_pandas()
 narrator_bios.loc[49845, 'Narrator Rank'] = 'رسول الله'
 narrator_bios.loc[49845, 'Number of Narrations'] = 0
 narrator_bios['Number of Narrations'] = narrator_bios['Number of Narrations'].astype(int)
 narrator_bios.loc[49845, 'Number of Narrations'] = 327512 
+narrator_bios['Generation'] = narrator_bios['Generation'].replace([None], [-1])
+narrator_bios['Generation'] = narrator_bios['Generation'].astype(int)
 
 
 
@@ -79,7 +81,7 @@ def network_narrator(narrator_id, fst_year, lst_year, yaxis):
     allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>"""
 
 def narrator_retriever(name):
-        return narrator_bios[(narrator_bios['Official Name'].str.contains(name)) | (narrator_bios['Famous Name'].str.contains(name)) | (narrator_bios['Rawi ID'].astype(str) == name)][['Rawi ID', 'Title Name', 'Official Name', 'Famous Name', 'Number of Narrations', 'Narrator Rank' ]]
+        return narrator_bios[(narrator_bios['Official Name'].str.contains(name)) | (narrator_bios['Famous Name'].str.contains(name)) | (narrator_bios['Rawi ID'].astype(str) == name)][['Rawi ID', 'Title Name', 'Official Name', 'Famous Name', 'Number of Narrations', 'Narrator Rank', 'Generation' ]]