Joshua Lochner commited on
Commit
6a8bf30
1 Parent(s): dccb47b

Update default classifier model

Browse files
Files changed (1) hide show
  1. src/model.py +1 -1
src/model.py CHANGED
@@ -116,7 +116,7 @@ class InferenceArguments(ModelArguments):
116
  }
117
  )
118
  classifier_model_name_or_path: str = field(
119
- default='Xenova/sponsorblock-classifier-v2',
120
  metadata={
121
  'help': 'Use a pretrained classifier'
122
  }
 
116
  }
117
  )
118
  classifier_model_name_or_path: str = field(
119
+ default='EColi/SB_Classifier',
120
  metadata={
121
  'help': 'Use a pretrained classifier'
122
  }