Update app.py

app.py

@@ -1,3 +1,25 @@
-import gradio as gr
+# import gradio as gr
 
-gr.load("models/Clinical-AI-Apollo/Medical-NER").launch()
+# gr.load("models/Clinical-AI-Apollo/Medical-NER").launch()
+import streamlit as st
+from transformers import pipeline
+
+pipe = pipeline("ner", model="Clinical-AI-Apollo/Medical-NER")
+
+def main():
+    text_input = st.text_area("Enter text:")
+
+    if text_input:
+        if text_input.strip() != "":
+            # Perform NER on the input text
+            entities = pipe(text_input)
+            st.write("Entities:")
+            for entity in entities:
+                st.write(f"- Entity: {entity['word']}, Type: {entity['entity']}")
+        else:
+            st.write("Please enter some text to extract entities.")
+
+if __name__ == "__main__":
+    main()
+
+pipe = pipeline()