Update app.py

app.py

@@ -1,9 +1,23 @@
 import streamlit as st
-from transformers import pipeline
+import mediapipe as mp
+#from transformers import pipeline
 
-pipe = pipeline("sentiment-analysis")
-text = st.text_area("enter some text")
+base_options = mp.tasks.BaseOptions(model_asset_path="classifier.tflite")
+options =      mp.tasks.text.TextClassifierOptions(base_options=base_options)
+with mp.tasks.text.TextClassifier.create_from_options(options) as classifier:
+    text = st.text_area("enter some text")
+    result = classifier.classify(text)
 
-if text:
-    out = pipe(text)
-    st.json(out)
+    category = classifications[0].categories[0]
+    if text and category:
+        st.json({
+            "name": category.category_name,
+            "score": category.score
+        })
+
+#pipe = pipeline("sentiment-analysis")
+#text = st.text_area("enter some text")
+
+#if text:
+#    out = pipe(text)
+#    st.json(out)