# decorators.py

from spaces import GPU  # Asegúrate de que 'spaces' está correctamente importado desde Hugging Face

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