marcelcastrobr commited on
Commit
c995fd5
1 Parent(s): a28c1a4

update of application

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -8,8 +8,11 @@ def sequence_to_classify(sequence, labels):
8
  clean_output = {idx: float(response['scores'].pop()) for idx in response['labels']}
9
 
10
  hypothesis_template = 'Dette eksempelet er {}.'
11
- response1 = classifier(sequence, labels, hypothesis_template=hypothesis_template, multi_class=True)
12
- clean_output1 = {idx: float(response1['scores'].pop()) for idx in response1['labels']}
 
 
 
13
 
14
  print("response is:{}".format(response))
15
  print(type(response))
 
8
  clean_output = {idx: float(response['scores'].pop()) for idx in response['labels']}
9
 
10
  hypothesis_template = 'Dette eksempelet er {}.'
11
+ label_clean = str(labels).split(",")
12
+ response1 = classifier(sequence, label_clean, hypothesis_template=hypothesis_template, multi_class=True)
13
+ labels = response1['labels']
14
+ scores = response1['scores']
15
+ clean_output1 = {idx: float(scores.pop(0)) for idx in label_clean}
16
 
17
  print("response is:{}".format(response))
18
  print(type(response))