Upload app.py

app.py

@@ -42,9 +42,6 @@ custom_css = r'''
 }
 '''
 
-# ========== 存储的路径 ==========
-print('HF_HOME', os.getenv("HF_HOME"))
-
 
 #  ========== 适配 SillyTavern 的模版 ==========
 def text_format(text: str, _env=None, **env):
@@ -111,21 +108,10 @@ with gr.Blocks() as setting:
         print(f"The file {setting_path.value} exists.")
     else:
         from huggingface_hub import snapshot_download
-
-
-        def hf_test():
-            directory = "/data/.huggingface/"
-            # 遍历目录下的所有文件
-            for root, dirs, files in os.walk(directory):
-                for file in files:
-                    print(os.path.join(root, file))
-
-
-        hf_test()
         os.mkdir("downloads")
         os.mkdir("cache")
-        tmp = snapshot_download(repo_id='TheBloke/CausalLM-7B-GGUF', allow_patterns='causallm_7b.Q5_K_M.gguf')
-        print('snapshot_download', tmp)
+        snapshot_download(repo_id='TheBloke/CausalLM-7B-GGUF', local_dir=r'downloads',
+                          allow_patterns='causallm_7b.Q5_K_M.gguf')
 
     #  ========== 加载模型 ==========
     model = StreamingLLM(model_path=setting_path.value,