spaces.py

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