del dir

app.py

@@ -2,9 +2,14 @@ import spaces
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import whoami
-import os
+import shutil
 
-os.removedirs('/data-nvme/zerogpu-offload/')
+try:
+    shutil.rmtree('/data-nvme/zerogpu-offload/')
+except FileNotFoundError:
+    print("目录不存在")
+except Exception as e:
+    print(f"删除目录失败：{e}")
 
 # 定义系统提示语
 system_prompt = """你是 Skywork-o1，Skywork AI 开发的思维模型，擅长通过深度思考解决涉及数学、编码和逻辑推理的复杂问题。面对用户请求时，你首先会进行一段漫长而深入的思考过程，探索问题的可能解决方案。完成思考后，你会在回复中详细解释解决过程。"""