Victoria Oberascher commited on
Commit
fb04593
·
1 Parent(s): 6219d1f

try to fix bug

Browse files
Files changed (1) hide show
  1. horizon-metrics.py +2 -0
horizon-metrics.py CHANGED
@@ -179,6 +179,8 @@ class HorizonMetrics(evaluate.Metric):
179
  # this means that the metric was called using the input field on huggingface website
180
 
181
  try:
 
 
182
  print(np.asarray(self.ground_truth_det).shape)
183
  print(np.asarray(self.predictions).shape)
184
  except:
 
179
  # this means that the metric was called using the input field on huggingface website
180
 
181
  try:
182
+ self.ground_truth_det = references
183
+ self.predictions = predictions
184
  print(np.asarray(self.ground_truth_det).shape)
185
  print(np.asarray(self.predictions).shape)
186
  except: