Update app.py

app.py

@@ -7,10 +7,13 @@ from datasets import load_dataset
 Secret_token = os.getenv('token')
 
 dataset = load_dataset('FDSRashid/hadith_info',data_files = 'Basic_Edge_Information.csv', token = Secret_token, split = 'train')
+dataset2 = load_dataset('FDSRashid/hadith_info',data_files = 'masteredgecityratiosapplied_updated_with_ID_mapping_to_NEO4J.csv', token = Secret_token, split = 'train')
 edge_info = dataset.to_pandas()
+taraf_info = dataset2.to_pandas()
+cities = taraf_info['City'].unique().tolist()
 
 def subset_city_year( city = ['المدينه', 'بغداد', 'كوفة', 'بصرة'], year = 50):
-  edges = edge_info[(edge_info['Year'] == year) & (edge_info['City'].isin(city))]
+  edges = edge_info[(taraf_info['Year'] == year) & (taraf_info['City'].isin(city))]
   return edges