Upload folder using huggingface_hub

app.py

@@ -5,6 +5,24 @@ from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
 import html
 
+# 导入 spaces 模块用于 GPU 检测
+is_spaces = os.getenv("SPACE_ID") is not None
+spaces_available = False
+GPU = None
+
+if is_spaces:
+    try:
+        from spaces import GPU
+        spaces_available = True
+    except ImportError:
+        print("⚠️ spaces module not available, GPU detection may not work")
+
+# 创建条件装饰器
+def gpu_decorator(func):
+    """条件应用 GPU 装饰器"""
+    if spaces_available and GPU is not None:
+        return GPU(func)
+    return func
 sys_prompt = """First output the the types of degradations in image briefly in <TYPE> <TYPE_END> tags, 
         and then output what effects do these degradation have on the image in <INFLUENCE> <INFLUENCE_END> tags, 
         then based on the strength of degradation, output an APPROPRIATE length for the reasoning process in <REASONING> <REASONING_END> tags, 
@@ -13,10 +31,8 @@ sys_prompt = """First output the the types of degradations in image briefly in <
 
 project_dir = os.path.dirname(os.path.abspath(__file__))
 
-is_spaces = os.getenv("SPACE_ID") is not None
-# temp_dir = os.path.join(project_dir, ".gradio_temp")
-temp_dir = project_dir
 if not is_spaces:
+    temp_dir = os.path.join(project_dir, ".gradio_temp")
     os.makedirs(temp_dir, exist_ok=True)
     os.environ["GRADIO_TEMP_DIR"] = temp_dir