caliangandrew commited on
Commit
18a48f3
·
verified ·
1 Parent(s): ccc7fee

Update test_detector.py

Browse files
Files changed (1) hide show
  1. test_detector.py +1 -1
test_detector.py CHANGED
@@ -38,7 +38,7 @@ class TestDetector(DeepfakeDetector):
38
  cuda (bool): Whether to enable cuda (GPU).
39
  """
40
 
41
- def __init__(self, model_name: str = 'UCF', config: str = 'ucf.yaml', cuda: bool = True):
42
  super().__init__(model_name, config, cuda)
43
 
44
  def ensure_weights_are_available(self, weight_filename):
 
38
  cuda (bool): Whether to enable cuda (GPU).
39
  """
40
 
41
+ def __init__(self, model_name: str = 'test', config: str = 'test_config.yaml', cuda: bool = True):
42
  super().__init__(model_name, config, cuda)
43
 
44
  def ensure_weights_are_available(self, weight_filename):