orionweller commited on
Commit
9121427
·
1 Parent(s): a667370
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,6 @@ class RepLlamaModel:
94
  model.eval()
95
  return model
96
 
97
- @spaces.GPU
98
  def encode(self, texts, batch_size=32, **kwargs):
99
  self.model = self.model.cuda()
100
  all_embeddings = []
@@ -181,6 +180,7 @@ def evaluate(qrels, results, k_values):
181
 
182
  return metrics
183
 
 
184
  def run_evaluation(dataset, postfix):
185
  global current_dataset, queries, model
186
  current_dataset = dataset
 
94
  model.eval()
95
  return model
96
 
 
97
  def encode(self, texts, batch_size=32, **kwargs):
98
  self.model = self.model.cuda()
99
  all_embeddings = []
 
180
 
181
  return metrics
182
 
183
+ @spaces.GPU
184
  def run_evaluation(dataset, postfix):
185
  global current_dataset, queries, model
186
  current_dataset = dataset