Hugging Face's logo

Spaces:
adamtayzzz
/
test
Running

App Files Community
adamtayzzz commited on
Commit
cab0f57
·
1 Parent(s): e7f8d89

Update whitebox_utils/classifier.py

Browse files
Files changed (1) hide show
  1. whitebox_utils/classifier.py +1 -1
whitebox_utils/classifier.py CHANGED
@@ -62,7 +62,7 @@ class MyClassifier():
62
 
63
  def set_exit_position(self,exit_pos):
64
  if self.model_type == 'albert':
65
- self.model.albert.set_exit_pos = exit_pos
66
 
67
  def reset_status(self,mode,stats=False):
68
  if self.model_type == 'albert':
 
62
 
63
  def set_exit_position(self,exit_pos):
64
  if self.model_type == 'albert':
65
+ self.model.albert.set_exit_pos(exit_pos)
66
 
67
  def reset_status(self,mode,stats=False):
68
  if self.model_type == 'albert':