Kaushik Bar
commited on
Commit
·
b5bfdc5
1
Parent(s):
7dd13e4
first commit
Browse files
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 |
-
|
|
|
|
|
|
|
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)
|