marcelcastrobr
commited on
Commit
·
c995fd5
1
Parent(s):
a28c1a4
update of application
Browse files
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 |
-
|
12 |
-
|
|
|
|
|
|
|
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))
|