decorators.py

# decorators.py

import spaces

def gpu_decorator(duration=100):
    def decorator(func):
        @spaces.GPU(duration=duration)
        def wrapper(*args, **kwargs):
            return func(*args, **kwargs)
        return wrapper
    return decorator