Update app.py

app.py

@@ -28,6 +28,8 @@ inverse_dict = {
     'material_of': 'made_from_material'
 }
 
+all_relations = ['uses', 'has_part', 'has_use', 'part_of', 'made_from_material', 'material_of']
+
 st.title("Materials use case search app")
 
 # User Input
@@ -337,7 +339,7 @@ if len(match_df) > 0:
                         st.write("Other inferences involving the same edge")
                         neg_instances[['source_en', 'relation', 'target_en', 'text', 'source_wikidata', 'target_wikidata', 'source', 'page_title', 'score', 'section']]
 
-                        count_dict = dict(Counter(instances.relation.to_list() + neg_instances.relation.to_list()))
+                        count_dict = {relation: rel_data[f'{relation}_support'] for relation in all_relations}
                         count_df = pd.DataFrame.from_dict(count_dict, orient='index')
                         count_df.columns = ['count']
                         st.write("Inference distribution for above edge")