Spaces:
Running
on
Zero
Running
on
Zero
Fix compatability for ZeroGPU
Browse files- profanity_detector.py +2 -0
profanity_detector.py
CHANGED
|
@@ -26,6 +26,8 @@ logger = logging.getLogger('profanity_detector')
|
|
| 26 |
|
| 27 |
# Detect if we're running in a ZeroGPU environment
|
| 28 |
IS_ZEROGPU = os.environ.get("SPACE_RUNTIME_STATELESS", "0") == "1"
|
|
|
|
|
|
|
| 29 |
|
| 30 |
# Define device strategy that works in both environments
|
| 31 |
if IS_ZEROGPU:
|
|
|
|
| 26 |
|
| 27 |
# Detect if we're running in a ZeroGPU environment
|
| 28 |
IS_ZEROGPU = os.environ.get("SPACE_RUNTIME_STATELESS", "0") == "1"
|
| 29 |
+
if os.environ.get("SPACES_ZERO_GPU") is not None:
|
| 30 |
+
IS_ZEROGPU = True
|
| 31 |
|
| 32 |
# Define device strategy that works in both environments
|
| 33 |
if IS_ZEROGPU:
|