Omkar008 commited on
Commit
78babf2
·
verified ·
1 Parent(s): e2387e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -50,9 +50,9 @@ async def process_document_base64(request: Request):
50
  if entities is not None:
51
  for ent in entities:
52
  if ent.get('type') is not None:
53
- mention_text = ent.get('mention_text')
54
- normalised_values = ent.get('normalizedValue') if 'normalizedValue' in entities else None
55
- document_entities[ent.get('type')] = {"mention_text":ent.get('mention_text'),"normalizedValue":normalised_values}
56
 
57
  return {"message_id":message_id , "filename":filename , "entities_data":document_entities}
58
 
 
50
  if entities is not None:
51
  for ent in entities:
52
  if ent.get('type') is not None:
53
+ mention_text = ent.get('mentionText')
54
+ normalised_values = ent.get('normalizedValue') if 'normalizedValue' in ent else None
55
+ document_entities[ent.get('type')] = {"mention_text":mention_text,"normalizedValue":normalised_values}
56
 
57
  return {"message_id":message_id , "filename":filename , "entities_data":document_entities}
58