Puree commited on
Commit
7f8feae
1 Parent(s): f54f989

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ Original file is located at
10
  import streamlit as st
11
  from transformers import pipeline
12
 
13
- classifier = pipeline("text-generation", model="Jean-Baptiste/camembert-ner")
14
  def main():
15
  st.title("Yelp review")
16
 
 
10
  import streamlit as st
11
  from transformers import pipeline
12
 
13
+ classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
14
  def main():
15
  st.title("Yelp review")
16