Update app.py

app.py

@@ -1,11 +1,23 @@
-import streamlit as st
+# import streamlit as st
 
+# from transformers import pipeline
+
+# pipe = pipeline("text-classification", model="Beehzod/smart-finetuned-ner")
+
+# text = st.text_area('enter text: ')
+
+# if text:
+#     out = pipe(text)
+#     st.json(out)
 from transformers import pipeline
 
-pipe = pipeline("text-classification", model="Beehzod/smart-finetuned-ner")
+# Load the model from the Hugging Face Hub
+ner_pipeline = pipeline("ner", model="Beehzod/smart-finetuned-ner")
 
-text = st.text_area('enter text: ')
+# Example predictions
+text = "Hugging Face Inc. is based in New York City, and its CEO is Clement Delangue."
+results = ner_pipeline(text)
 
-if text:
-    out = pipe(text)
-    st.json(out)
+for entity in results:
+    #print(f"Entity: {entity['word']}, Label: {entity['entity']}, Score: {entity['score']:.4f}")
+    st.json(entity)