blazingbunny commited on
Commit
82bcffd
·
verified ·
1 Parent(s): 98b7999

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -23,6 +23,10 @@ def count_entities(entities):
23
  count += 1
24
  return count
25
 
 
 
 
 
26
  # Function to export entities as a JSON or CSV file
27
  def export_entities(entities):
28
  entity_list = []
@@ -31,7 +35,7 @@ def export_entities(entities):
31
  "Name": entity.name,
32
  "Type": language_v1.Entity.Type(entity.type_).name,
33
  "Salience Score": entity.salience,
34
- "Metadata": entity.metadata,
35
  "Mentions": [mention.text.content for mention in entity.mentions]
36
  }
37
  entity_list.append(entity_info)
 
23
  count += 1
24
  return count
25
 
26
+ # Function to serialize entity metadata
27
+ def serialize_entity_metadata(metadata):
28
+ return {k: str(v) for k, v in metadata.items()}
29
+
30
  # Function to export entities as a JSON or CSV file
31
  def export_entities(entities):
32
  entity_list = []
 
35
  "Name": entity.name,
36
  "Type": language_v1.Entity.Type(entity.type_).name,
37
  "Salience Score": entity.salience,
38
+ "Metadata": serialize_entity_metadata(entity.metadata),
39
  "Mentions": [mention.text.content for mention in entity.mentions]
40
  }
41
  entity_list.append(entity_info)