src/gpu_decorator.py

"""ZeroGPU compatibility decorator for HuggingFace Spaces."""

try:
    import spaces
    GPU = spaces.GPU
except ImportError:
    def GPU(func=None, duration=None):
        if func is None:
            return lambda f: f
        return func