Update app.py

app.py

@@ -8,7 +8,7 @@ st.set_page_config(layout="wide")
 
 @st.cache(allow_output_mutation = True)
 def get_classifier_model():
-    return pipeline("zero-shot-classification", model="models/bart-large-mnli")
+    return pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
     #return pipeline("zero-shot-classification",model="sentence-transformers/paraphrase-MiniLM-L6-v2")
 
 
@@ -24,7 +24,7 @@ st.markdown("***")
 
 col1, col2, col3 = st.columns((1,1,1))
 col1.header("Select Sentiments")
-sentiments = col1.multiselect("",["Happy","Sad","Anxious","Depressed","Empathetic"],["Happy","Sad","Anxious","Depressed","Empathetic"])
+sentiments = col1.multiselect("",["Happy","Sad","Neutral"])
 col2.header("Select Entities")
 entities = col2.multiselect("",["Employee","Doctor","Family","Friends"],
                             ["Employee","Doctor","Family","Friends"])