Spaces:
Running on Zero
Running on Zero
Lord-Raven commited on
Commit ·
249b712
1
Parent(s): abbb119
Trying zero-shot classification.
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ def zero_shot_classification(data_string):
|
|
| 7 |
print(data_string)
|
| 8 |
data = json.loads(data_string)
|
| 9 |
print(data)
|
| 10 |
-
classifier = pipeline(task='zero-shot-classification',
|
| 11 |
|
| 12 |
results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
|
| 13 |
return results
|
|
|
|
| 7 |
print(data_string)
|
| 8 |
data = json.loads(data_string)
|
| 9 |
print(data)
|
| 10 |
+
classifier = pipeline(task='zero-shot-classification', model='typeform/distilbert-base-uncased-mnli')
|
| 11 |
|
| 12 |
results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
|
| 13 |
return results
|