Kaushik Bar
commited on
Commit
·
b3cae23
1
Parent(s):
92be70e
first commit
Browse files
app.py
CHANGED
@@ -104,8 +104,7 @@ def sequence_to_classify(text, labels):
|
|
104 |
labels_features = torch.tensor(self.model.encode(labels))
|
105 |
|
106 |
sim_scores = util.cos_sim(text_features, labels_features)
|
107 |
-
|
108 |
-
|
109 |
for textlet, sim_score in zip([text], sim_scores):
|
110 |
out = []
|
111 |
pred = {}
|
|
|
104 |
labels_features = torch.tensor(self.model.encode(labels))
|
105 |
|
106 |
sim_scores = util.cos_sim(text_features, labels_features)
|
107 |
+
preds = []
|
|
|
108 |
for textlet, sim_score in zip([text], sim_scores):
|
109 |
out = []
|
110 |
pred = {}
|