Spaces:
Sleeping
Sleeping
Davide Fiocco
commited on
Commit
·
501b061
1
Parent(s):
38f2cf5
Switch to bart-large for inference
Browse files
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="
|
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")
|