Sasidhar commited on
Commit
c551049
1 Parent(s): a07631a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,7 +8,8 @@ st.set_page_config(layout="wide")
8
 
9
  @st.cache(allow_output_mutation = True)
10
  def get_classifier_model():
11
- return pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
12
  #return pipeline("zero-shot-classification",model="sentence-transformers/paraphrase-MiniLM-L6-v2")
13
 
14
  #st.sidebar.image("Suncorp-Bank-logo.png",width=255)
8
 
9
  @st.cache(allow_output_mutation = True)
10
  def get_classifier_model():
11
+ return pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")
12
+ #return pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
13
  #return pipeline("zero-shot-classification",model="sentence-transformers/paraphrase-MiniLM-L6-v2")
14
 
15
  #st.sidebar.image("Suncorp-Bank-logo.png",width=255)