Spaces:
Sleeping
Sleeping
import streamlit as st | |
import mediapipe as mp | |
#from transformers import pipeline | |
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) | |
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) | |