Spaces:

Donlapark
/

sample-text-classification

Sleeping

Donlapark commited on Oct 4, 2022

Commit

066162b

•

1 Parent(s): 67aff2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
 import streamlit as st
 from transformers import pipeline
-classifier = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
 def main():
-    st.title("Sentiment analysis")
     with st.form("text_field"):
         text = st.text_area('enter some text:')

 import streamlit as st
 from transformers import pipeline
+classifier = pipeline("sentiment-analysis", model="Donlapark/finetuned_yelp")
 def main():
+    st.title("Yelp review")
     with st.form("text_field"):
         text = st.text_area('enter some text:')