Upload app.py

app.py

@@ -103,6 +103,16 @@ with gr.Blocks() as setting:
         setting_mirostat_eta = gr.Number(label="Mirostat 学习率", interactive=True, **cfg['setting_mirostat_eta'])
         setting_mirostat_tau = gr.Number(label="Mirostat 目标熵", interactive=True, **cfg['setting_mirostat_tau'])
 
+    #  ========== 下载模型 ==========
+    if os.path.exists(setting_path.value):
+        print(f"The file {setting_path.value} exists.")
+    else:
+        from huggingface_hub import snapshot_download
+        os.mkdir("downloads")
+        os.mkdir("cache")
+        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,
                          seed=setting_seed.value,