fathyshalab commited on
Commit
ba922c0
·
1 Parent(s): 103b350

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ classifier = pipeline("zero-shot-classification",
7
  model="joeddav/xlm-roberta-large-xnli",use_auth_token=token)
8
 
9
  def classify(text):
10
- return classifier(text, candidate_labels, multi_label=True)
11
 
12
  candidate_labels = ["wasser", "strom", "gas", "hausreinigung", "schoenheit", "wellness", "oeffentlichkeit", "oeffentlicher", "verkehr", "vermietung", "medizin", "gesundheit", "pflege", "reisen", "tourismus", "versicherungen", "recht"]
13
 
 
7
  model="joeddav/xlm-roberta-large-xnli",use_auth_token=token)
8
 
9
  def classify(text):
10
+ return classifier(text, candidate_labels)
11
 
12
  candidate_labels = ["wasser", "strom", "gas", "hausreinigung", "schoenheit", "wellness", "oeffentlichkeit", "oeffentlicher", "verkehr", "vermietung", "medizin", "gesundheit", "pflege", "reisen", "tourismus", "versicherungen", "recht"]
13