decorators.py

# decorators.py

import spaces

# Decorador personalizado que usa GPU si está disponible
def gpu_decorator(duration=100):
    def decorator(func):
        @spaces.GPU(duration=duration)
        def wrapper(*args, **kwargs):
            return func(*args, **kwargs)
        return wrapper
    return decorator