Update app.py

app.py

@@ -28,6 +28,13 @@ import cv2
 import requests
 import re
 from PIL import Image
+import streamlit as st
+import os
+
+key=os.environ.getattribute("api_key")
+print(key)
+API_URL = "https://api-inference.huggingface.co/models/flair/ner-english-large"
+headers = {"Authorization": "Bearer {key}"}
 
 ## Image uploading function ##
 def image_upload_and_ocr(reader):
@@ -42,6 +49,10 @@ def image_upload_and_ocr(reader):
     return result,result2
 
 
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
         
     ###  DRAWING DETECTION FUNCTION  ###
 def drawing_detection(image):
@@ -71,4 +82,25 @@ 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)}")
+
+