Kaushik Bar commited on
Commit
b3cae23
·
1 Parent(s): 92be70e

first commit

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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
- preds = []
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 = {}