|
|
|
"""Lab10app.ipynb |
|
|
|
Automatically generated by Colaboratory. |
|
|
|
Original file is located at |
|
https://colab.research.google.com/drive/1lrMgwkvs4N_RUAAPjhRcbd1shmVd38Ah |
|
""" |
|
|
|
import streamlit as st |
|
from transformers import pipeline |
|
|
|
classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base") |
|
def main(): |
|
st.title("Emotion(ENG) classification") |
|
|
|
with st.form("text_field"): |
|
text = st.text_area('enter your text:') |
|
|
|
clicked = st.form_submit_button("Submit") |
|
if clicked: |
|
results = classifier([text]) |
|
st.json(results) |
|
|
|
if __name__ == "__main__": |
|
main() |
