kaushikbar commited on
Commit
3861998
1 Parent(s): 3692b96

added multi-label as a radio button user input

Browse files
Files changed (1) hide show
  1. app.py +24 -20
app.py CHANGED
@@ -93,22 +93,22 @@ def prep_examples():
93
  example_labels16 = "Under the leadership of Nadella;;Despite the leadership of Nadella;;And Nadella had nothing to do with it;;Under the leadership of Pichai"
94
 
95
  examples = [
96
- [example_text1, example_labels1],
97
- [example_text2, example_labels2],
98
- [example_text3, example_labels3],
99
- [example_text4, example_labels4],
100
- [example_text5, example_labels5],
101
- [example_text6, example_labels6],
102
- [example_text7, example_labels7],
103
- [example_text8, example_labels8],
104
- [example_text9, example_labels9],
105
- [example_text10, example_labels10],
106
- [example_text11, example_labels11],
107
- [example_text12, example_labels12],
108
- [example_text13, example_labels13],
109
- [example_text14, example_labels14],
110
- [example_text15, example_labels15],
111
- [example_text16, example_labels16]]
112
 
113
  return examples
114
 
@@ -154,12 +154,13 @@ def detect_lang(sequence, labels):
154
 
155
  return seq_lang
156
 
157
- def sequence_to_classify(sequence, labels):
158
  lang = 'en' #detect_lang(sequence, labels)
159
  classifier = classifiers[lang]
160
 
161
  label_clean = str(labels).split(";;")
162
- response = classifier(sequence, label_clean, multi_label=False)
 
163
 
164
  predicted_labels = response['labels']
165
  predicted_scores = response['scores']
@@ -169,6 +170,7 @@ def sequence_to_classify(sequence, labels):
169
  sequence,
170
  predicted_labels))
171
 
 
172
  return clean_output
173
 
174
  iface = gr.Interface(
@@ -180,8 +182,10 @@ iface = gr.Interface(
180
  placeholder="Text here..."),
181
  gr.inputs.Textbox(lines=2,
182
  label="Please enter the candidate labels (separated by 2 consecutive semicolons)...",
183
- placeholder="Labels here separated by ;;")],
184
- outputs=gr.outputs.Label(num_top_classes=1),
 
 
185
  #interpretation="default",
186
  examples=prep_examples())
187
 
 
93
  example_labels16 = "Under the leadership of Nadella;;Despite the leadership of Nadella;;And Nadella had nothing to do with it;;Under the leadership of Pichai"
94
 
95
  examples = [
96
+ [example_text1, example_labels1, False],
97
+ [example_text2, example_labels2, False],
98
+ [example_text3, example_labels3, False],
99
+ [example_text4, example_labels4, True],
100
+ [example_text5, example_labels5, True],
101
+ [example_text6, example_labels6, False],
102
+ [example_text7, example_labels7, False],
103
+ [example_text8, example_labels8, False],
104
+ [example_text9, example_labels9, True],
105
+ [example_text10, example_labels10, False],
106
+ [example_text11, example_labels11, False],
107
+ [example_text12, example_labels12, False],
108
+ [example_text13, example_labels13, False],
109
+ [example_text14, example_labels14, False],
110
+ [example_text15, example_labels15, False],
111
+ [example_text16, example_labels16, False]]
112
 
113
  return examples
114
 
 
154
 
155
  return seq_lang
156
 
157
+ def sequence_to_classify(sequence, labels, multi_label):
158
  lang = 'en' #detect_lang(sequence, labels)
159
  classifier = classifiers[lang]
160
 
161
  label_clean = str(labels).split(";;")
162
+ print("**** multi_label: ", multi_label)
163
+ response = classifier(sequence, label_clean, multi_label=multi_label)
164
 
165
  predicted_labels = response['labels']
166
  predicted_scores = response['scores']
 
170
  sequence,
171
  predicted_labels))
172
 
173
+ print("++++clean_output: ", clean_output)
174
  return clean_output
175
 
176
  iface = gr.Interface(
 
182
  placeholder="Text here..."),
183
  gr.inputs.Textbox(lines=2,
184
  label="Please enter the candidate labels (separated by 2 consecutive semicolons)...",
185
+ placeholder="Labels here separated by ;;"),
186
+ gr.inputs.Textbox(choices=[False, True],
187
+ label="Multi-label?"],
188
+ outputs=gr.outputs.Label(num_top_classes=5),
189
  #interpretation="default",
190
  examples=prep_examples())
191