Davide Fiocco commited on
Commit
501b061
·
1 Parent(s): 38f2cf5

Switch to bart-large for inference

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
 
5
  st.set_page_config(page_title="Zero-shot classification from tabular data", page_icon=None, layout="wide", initial_sidebar_state="auto", menu_items=None)
6
 
7
- classifier = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")
8
 
9
  st.title("Zero-shot classification from tabular data")
10
  st.text("Upload an Excel table and perform zero-shot classification on a set of custom labels")
 
4
 
5
  st.set_page_config(page_title="Zero-shot classification from tabular data", page_icon=None, layout="wide", initial_sidebar_state="auto", menu_items=None)
6
 
7
+ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
8
 
9
  st.title("Zero-shot classification from tabular data")
10
  st.text("Upload an Excel table and perform zero-shot classification on a set of custom labels")