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

update of application

Browse files
Files changed (1) hide show
  1. app.py +6 -11
app.py CHANGED
@@ -9,20 +9,14 @@ 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
  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."]
22
- example_labels=["politikk,helse,sport,religion", "helse,sport,religion, mat"]
23
-
24
- def greet(name):
25
- return "Hello " + name + "!!"
26
 
27
  iface = gr.Interface(
28
  title = "Zero-shot Classification of Norwegian Text",
@@ -38,6 +32,7 @@ iface = gr.Interface(
38
  capture_session=True,
39
  interpretation="default"
40
  ,examples=[
41
- [example_text, example_labels]
 
42
  ])
43
  iface.launch()
 
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 = {idx: float(predicted_scores.pop(0)) for idx in predicted_labels}
13
 
 
 
 
 
14
  return clean_output
15
 
16
+ example_text1="Folkehelseinstituttets mest optimistiske anslag er at alle voksne er ferdigvaksinert innen midten av september."
17
+ example_labels1="politikk,helse,sport,religion"
18
+ example_text2="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."
19
+ example_labels2="helse,sport,religion, mat"
 
20
 
21
  iface = gr.Interface(
22
  title = "Zero-shot Classification of Norwegian Text",
 
32
  capture_session=True,
33
  interpretation="default"
34
  ,examples=[
35
+ [example_text1, example_labels1],
36
+ [example_text2, example_labels2]
37
  ])
38
  iface.launch()