Kaushik Bar commited on
Commit
7bbe89d
·
1 Parent(s): b5bfdc5

first commit

Browse files
Files changed (1) hide show
  1. app.py +16 -16
app.py CHANGED
@@ -56,17 +56,17 @@ def prep_examples():
56
  example_labels11 = "bhalo;;kharap"
57
 
58
  examples = [
59
- [example_text1, example_labels1],
60
- [example_text2, example_labels2],
61
- [example_text3, example_labels3],
62
- [example_text4, example_labels4],
63
- [example_text5, example_labels5],
64
- [example_text6, example_labels6],
65
- [example_text7, example_labels7],
66
- [example_text8, example_labels8],
67
- [example_text9, example_labels9],
68
- [example_text10, example_labels10],
69
- [example_text11, example_labels11]]
70
 
71
  return examples
72
 
@@ -92,10 +92,10 @@ def sequence_to_classify(text, labels, hypothesis_template):
92
  else:
93
  model = model_multi
94
 
95
- if hypothesis_template:
96
- labels = [hypothesis_template.format(label) for label in labels.split(";;")]
97
- else:
98
- labels = [label for label in labels.split(";;")]
99
 
100
  if str(type(model)) == "<class 'clip.model.CLIP'>":
101
  text_tokens = clip.tokenize(text)
@@ -151,7 +151,7 @@ iface = gr.Interface(
151
  label="Please enter the candidate labels (separated by 2 consecutive semicolons)...",
152
  placeholder="Labels here separated by ;;"),
153
  gr.inputs.Textbox(lines=2,
154
- label="Please enter the text for hypothesis template(optional)...",
155
  placeholder="Text here...")],
156
  outputs=gr.outputs.Label(num_top_classes=5),
157
  #interpretation="default",
 
56
  example_labels11 = "bhalo;;kharap"
57
 
58
  examples = [
59
+ [example_text1, example_labels1, ""],
60
+ [example_text2, example_labels2, ""],
61
+ [example_text3, example_labels3, ""],
62
+ [example_text4, example_labels4, "The sentiment expressed here is {}."],
63
+ [example_text5, example_labels5, ""],
64
+ [example_text6, example_labels6, ""],
65
+ [example_text7, example_labels7, ""],
66
+ [example_text8, example_labels8, ""],
67
+ [example_text9, example_labels9, ""],
68
+ [example_text10, example_labels10, ""],
69
+ [example_text11, example_labels11, ""]]
70
 
71
  return examples
72
 
 
92
  else:
93
  model = model_multi
94
 
95
+ if hypothesis_template == None:
96
+ hypothesis_template = "{}"
97
+ print("I'm here")
98
+ labels = [hypothesis_template.format(label) for label in labels.split(";;")]
99
 
100
  if str(type(model)) == "<class 'clip.model.CLIP'>":
101
  text_tokens = clip.tokenize(text)
 
151
  label="Please enter the candidate labels (separated by 2 consecutive semicolons)...",
152
  placeholder="Labels here separated by ;;"),
153
  gr.inputs.Textbox(lines=2,
154
+ label="Please enter the text for hypothesis template (optional)...",
155
  placeholder="Text here...")],
156
  outputs=gr.outputs.Label(num_top_classes=5),
157
  #interpretation="default",