Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
 
 # โหลด Tokenizer และ Model
@@ -9,11 +9,11 @@ model = AutoModelForTokenClassification.from_pretrained(model_name)
 # สร้าง NER Pipeline
 ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
 
-# ฟังก์ชันที่รับข้อความและทำการวิเคราะห์
-def analyze_text(text):
-    ner_results = ner_pipeline(text)
-    return [(entity['word'], entity['entity'], entity['score']) for entity in ner_results]
+# UI ด้วย Streamlit
+st.title("NER Analysis App")
+text = st.text_area("Enter text for NER analysis:")
 
-# สร้าง UI ด้วย Gradio
-iface = gr.Interface(fn=analyze_text, inputs="text", outputs="dataframe", title="NER Analysis App")
-iface.launch()
+if st.button("Analyze"):
+    ner_results = ner_pipeline(text)
+    for entity in ner_results:
+        st.write(f"Entity: {entity['word']}, Label: {entity['entity']}, Score: {entity['score']:.4f}")