Update app.py

app.py

@@ -55,8 +55,10 @@ LOADER_MAPPING = {
 
 repo_name = "IlyaGusev/saiga2_13b_gguf"
 model_name = "model-q4_K.gguf"
+final_model_path = f"/data/{model_name}"
 embedder_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
 
+print("Downloading all files...")
 rm_files = [os.path.join("/data", f) for f in os.listdir("/data")]
 for f in rm_files:
    if os.path.isfile(f):
@@ -64,13 +66,13 @@ for f in rm_files:
    else:
        shutil.rmtree(f)
 
-print("Downloading all files...")
-snapshot_download(repo_id=repo_name, local_dir="/data", allow_patterns=model_name, local_dir_use_symlinks=False)
+if not os.path.exists(final_model_path):
+    snapshot_download(repo_id=repo_name, local_dir="/data", allow_patterns=model_name, local_dir_use_symlinks=False, force_download=True)
 # snapshot_download(repo_id=embedder_name, local_dir="/data/paraphrase-multilingual-mpnet-base-v2", cache_dir="/data")
 print("Files downloaded!")
 
 model = Llama(
-    model_path=f"/data/{model_name}",
+    model_path=final_model_path,
     n_ctx=2000,
     n_parts=1,
 )