kyled commited on
Commit
92518e0
1 Parent(s): a0667b7

added multi_class to model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ import pandas as pd
5
 
6
  st.title("Phrase Feeling Analysis")
7
  classifier = t.pipeline("zero-shot-classification",
8
- model="facebook/bart-large-mnli")
 
9
 
10
  x = st.text_input("Enter your title here:")
11
 
 
5
 
6
  st.title("Phrase Feeling Analysis")
7
  classifier = t.pipeline("zero-shot-classification",
8
+ model="facebook/bart-large-mnli",
9
+ multi_class=True)
10
 
11
  x = st.text_input("Enter your title here:")
12