likewendy commited on
Commit
9961c4d
1 Parent(s): 959e3d9
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -2,14 +2,9 @@ import spaces
2
  import gradio as gr
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  from huggingface_hub import whoami
5
- import shutil
6
 
7
- try:
8
- shutil.rmtree('/data-nvme/zerogpu-offload/')
9
- except FileNotFoundError:
10
- print("目录不存在")
11
- except Exception as e:
12
- print(f"删除目录失败:{e}")
13
 
14
  # 定义系统提示语
15
  system_prompt = """你是 Skywork-o1,Skywork AI 开发的思维模型,擅长通过深度思考解决涉及数学、编码和逻辑推理的复杂问题。面对用户请求时,你首先会进行一段漫长而深入的思考过程,探索问题的可能解决方案。完成思考后,你会在回复中详细解释解决过程。"""
 
2
  import gradio as gr
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  from huggingface_hub import whoami
5
+ import os
6
 
7
+ os.system("rm -rf /data-nvme/zerogpu-offload/*")
 
 
 
 
 
8
 
9
  # 定义系统提示语
10
  system_prompt = """你是 Skywork-o1,Skywork AI 开发的思维模型,擅长通过深度思考解决涉及数学、编码和逻辑推理的复杂问题。面对用户请求时,你首先会进行一段漫长而深入的思考过程,探索问题的可能解决方案。完成思考后,你会在回复中详细解释解决过程。"""