sileod commited on
Commit
72f30d6
·
verified ·
1 Parent(s): b92a5dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=text_pair='there is a black cat')],return_all_scores=True)
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, ''