diff --git a/packages/sharing-editor/public/samples/023_transformers/app.py b/packages/sharing-editor/public/samples/023_transformers/app.py new file mode 100644 index 000000000..9157635c0 --- /dev/null +++ b/packages/sharing-editor/public/samples/023_transformers/app.py @@ -0,0 +1,15 @@ +import streamlit as st + +from transformers_js_py import pipeline + +st.title("Sentiment analysis") + +text = st.text_input("Input some text", "I love transformers!") + +if text: + with st.spinner(): + if "pipe" not in st.session_state: + st.session_state["pipe"] = await pipeline('sentiment-analysis') + pipe = st.session_state["pipe"] + out = await pipe(text) + st.write(out) diff --git a/packages/sharing-editor/public/samples/023_transformers/requirements.txt b/packages/sharing-editor/public/samples/023_transformers/requirements.txt new file mode 100644 index 000000000..3702d6774 --- /dev/null +++ b/packages/sharing-editor/public/samples/023_transformers/requirements.txt @@ -0,0 +1 @@ +transformers-js-py diff --git a/packages/sharing-editor/public/samples/023_transformers/stlite.json b/packages/sharing-editor/public/samples/023_transformers/stlite.json new file mode 100644 index 000000000..9fa599ba4 --- /dev/null +++ b/packages/sharing-editor/public/samples/023_transformers/stlite.json @@ -0,0 +1,4 @@ +{ + "title": "Transformers", + "entrypoint": "app.py" +}