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

first commit

Browse files
Files changed (1) hide show
  1. app.py +15 -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
 
@@ -87,16 +87,15 @@ def detect_lang(text):
87
 
88
  def sequence_to_classify(text, labels, hypothesis_template):
89
  lang = detect_lang(text)
90
-
91
- print("hypothesis_template", hypothesis_template)
92
  if lang == 'en':
93
  model = model_en
94
- hypothesis_template = "{}"
95
  else:
96
  model = model_multi
97
- hypothesis_template = "{}"
98
 
99
- labels = [hypothesis_template.format(label) for label in labels.split(";;")]
 
 
 
100
 
101
  if str(type(model)) == "<class 'clip.model.CLIP'>":
102
  text_tokens = clip.tokenize(text)
 
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
 
 
87
 
88
  def sequence_to_classify(text, labels, hypothesis_template):
89
  lang = detect_lang(text)
 
 
90
  if lang == 'en':
91
  model = model_en
 
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)