diff --git a/deepfake.py b/deepfake.py index 0b8e8b8..023b238 100644 --- a/deepfake.py +++ b/deepfake.py @@ -19,7 +19,7 @@ log = logging.getLogger() @register(output_schema=["scorelist"], flag=OperatorFlag.STATELESS | OperatorFlag.REUSEABLE) -class Deepfakevit(Operator): +class deepfake(Operator): ''' ViT Distillation '''