Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def process_input(text_input, labels_or_premise, mode):
|
|
32 |
results = {label: score for label, score in zip(prediction['labels'], prediction['scores'])}
|
33 |
return results, ''
|
34 |
else: # NLI mode
|
35 |
-
pred=
|
36 |
results= {pred['label']:pred['score'] for pred in pred}
|
37 |
|
38 |
return results, ''
|
|
|
32 |
results = {label: score for label, score in zip(prediction['labels'], prediction['scores'])}
|
33 |
return results, ''
|
34 |
else: # NLI mode
|
35 |
+
pred= nli_classifier([{"text": text_input, "text_pair": labels_or_premise}],return_all_scores=True)[0]
|
36 |
results= {pred['label']:pred['score'] for pred in pred}
|
37 |
|
38 |
return results, ''
|