Update app.py

app.py

@@ -32,16 +32,17 @@ def ner(text: str):
             "score": ent.get("score", 0),
         }
         for ent in entities
-        if ent["label"] == "Textile Machinery"  # Exact match with capitalization
+        if ent["label"] == "Textile Machinery"
     ]
     
     # Prepare entities for color-coded display using gr.HighlightedText
     highlights = [(ent["start"], ent["end"], ent["entity"]) for ent in textile_entities]
     
+    # Return two outputs: one for the highlighted text and one for the entities in JSON format
     return {
         "text": text,
         "entities": highlights
-    }
+    }, textile_entities  # First output is for the highlighted text, second for the JSON component
 
 with gr.Blocks(title="Textile Machinery NER Demo") as demo:
     gr.Markdown(