import functools import os def GPU(func): """ A decorator to indicate that a function should use GPU acceleration if available. This is used specifically for Hugging Face Spaces. """ @functools.wraps(func) def wrapper(*args, **kwargs): return func(*args, **kwargs) return wrapper