marcelcastrobr commited on
Commit
3163c0c
1 Parent(s): 13a9362

update of application

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -9,10 +9,13 @@ def sequence_to_classify(sequence, labels):
9
  response = classifier(sequence, label_clean, hypothesis_template=hypothesis_template, multi_class=True)
10
  predicted_labels = response['labels']
11
  predicted_scores = response['scores']
12
- clean_output = {labels[idx]: float(predicted_scores[idx]) for idx in range(len(predicted_labels))}
 
13
 
14
  print("response is:{}".format(response))
15
  print("clean_output: {}".format(clean_output))
 
 
16
  return clean_output
17
 
18
  example_text=["Folkehelseinstituttets mest optimistiske anslag er at alle voksne er ferdigvaksinert innen midten av september.","Kutt smør i terninger, og la det temperere seg litt mens deigen elter. Ha hvetemel, sukker, gjær, salt og kardemomme i en bakebolle til kjøkkenmaskin. Bruker du fersk gjær kan du smuldre gjæren i bollen, eller røre den ut i melken. Alt vil ettehvert blande seg godt, så begge deler er like bra."]
 
9
  response = classifier(sequence, label_clean, hypothesis_template=hypothesis_template, multi_class=True)
10
  predicted_labels = response['labels']
11
  predicted_scores = response['scores']
12
+ #clean_output = {labels[idx]: float(predicted_scores[idx]) for idx in range(len(predicted_labels))}
13
+ clean_output = {idx: float(predicted_scores.pop(0)) for idx in predicted_labels}
14
 
15
  print("response is:{}".format(response))
16
  print("clean_output: {}".format(clean_output))
17
+ print(predicted_labels)
18
+ print(predicted_scores)
19
  return clean_output
20
 
21
  example_text=["Folkehelseinstituttets mest optimistiske anslag er at alle voksne er ferdigvaksinert innen midten av september.","Kutt smør i terninger, og la det temperere seg litt mens deigen elter. Ha hvetemel, sukker, gjær, salt og kardemomme i en bakebolle til kjøkkenmaskin. Bruker du fersk gjær kan du smuldre gjæren i bollen, eller røre den ut i melken. Alt vil ettehvert blande seg godt, så begge deler er like bra."]