anisaallahdadi commited on
Commit
f408130
·
1 Parent(s): 831a06a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,6 +1,11 @@
 
 
 
 
 
1
  from transformers import pipeline
2
  classifier = pipeline("zero-shot-classification",
3
- model="facebook/bart-large-mnli")
4
 
5
  def classify(input_query, input_classes, input_multi_class):
6
  result = []
 
1
+
2
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
+ model = AutoModelForSequenceClassification.from_pretrained('facebook/bart-large-mnli')
4
+ tokenizer = AutoTokenizer.from_pretrained('facebook/bart-large-mnli')
5
+
6
  from transformers import pipeline
7
  classifier = pipeline("zero-shot-classification",
8
+ model=model, tokenizer=tokenizer)
9
 
10
  def classify(input_query, input_classes, input_multi_class):
11
  result = []