Spaces:

tanlocc
/

NSFW_Check

Runtime error

tanlocc commited on Sep 27, 2023

Commit

4dc7058

1 Parent(s): aaeb498

Update base.py

Files changed (1) hide show

base.py CHANGED Viewed

@@ -10,7 +10,7 @@ from abc import ABC, abstractclassmethod
 __dir__ = Path(__file__).parent
 class ONNXBaseTask(ABC):
-    num_gpus: int = count_gpus()
     def __init__(self, weight: str) -> None:
         self.session = self.initialize_session(weight)

 __dir__ = Path(__file__).parent
 class ONNXBaseTask(ABC):
+    num_gpus: int = 0
     def __init__(self, weight: str) -> None:
         self.session = self.initialize_session(weight)