Update app.py

app.py

@@ -53,6 +53,24 @@ def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
+def get_ner_from_transformer(output):
+    data = output
+    named_entities = {}
+    for entity in data:
+        entity_type = entity['entity_group']
+        entity_text = entity['word']
+        
+        if entity_type not in named_entities:
+            named_entities[entity_type] = []
+        
+        named_entities[entity_type].append(entity_text)
+    
+    for entity_type, entities in named_entities.items():
+        # print(f"{entity_type}: {', '.join(entities)}")
+    return entity_type,named_entities
+    
+        
+
         
     ###  DRAWING DETECTION FUNCTION  ###
 def drawing_detection(image):
@@ -79,28 +97,11 @@ st.title("_Business_ card data extractor using opencv and streamlit :sunglasses:
 result,result2=image_upload_and_ocr(reader)
 darwing_image=drawing_detection(result2)
 
-
-
-    
+ 
 output = query({
     "inputs": result,
 })
 
-data = output
-
-named_entities = {}
-
-for entity in data:
-    entity_type = entity['entity_group']
-    entity_text = entity['word']
-    
-    if entity_type not in named_entities:
-        named_entities[entity_type] = []
-    
-    named_entities[entity_type].append(entity_text)
-
-for entity_type, entities in named_entities.items():
-    print(f"{entity_type}: {', '.join(entities)}")
-
-    
-    
+entity_type,named_entities= get_ner_from_transformer(output)
+st.write(entity_type)
+st.write(named_entities)