Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ from lime import lime_text
|
|
| 22 |
|
| 23 |
stopwords_list = stopwords.words('english') + ['your_additional_stopwords_here']
|
| 24 |
st.set_page_config(layout="wide")
|
| 25 |
-
|
| 26 |
def load_model_and_tokenizer(model_name):
|
| 27 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 28 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
@@ -30,7 +30,7 @@ def load_model_and_tokenizer(model_name):
|
|
| 30 |
|
| 31 |
model, tokenizer = load_model_and_tokenizer('nlptown/bert-base-multilingual-uncased-sentiment')
|
| 32 |
|
| 33 |
-
|
| 34 |
def load_pipeline():
|
| 35 |
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
|
| 36 |
return classifier
|
|
|
|
| 22 |
|
| 23 |
stopwords_list = stopwords.words('english') + ['your_additional_stopwords_here']
|
| 24 |
st.set_page_config(layout="wide")
|
| 25 |
+
|
| 26 |
def load_model_and_tokenizer(model_name):
|
| 27 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 28 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
|
| 30 |
|
| 31 |
model, tokenizer = load_model_and_tokenizer('nlptown/bert-base-multilingual-uncased-sentiment')
|
| 32 |
|
| 33 |
+
|
| 34 |
def load_pipeline():
|
| 35 |
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
|
| 36 |
return classifier
|