IlyaGusev commited on
Commit
ce7d64f
1 Parent(s): 45a6f55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -59,7 +59,10 @@ embedder_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
59
 
60
  rm_files = [os.path.join("/data", f) for f in os.listdir("/data")]
61
  for f in rm_files:
62
- shutil.rmtree(f)
 
 
 
63
 
64
  print("Downloading all files...")
65
  snapshot_download(repo_id=repo_name, local_dir="/data/saiga_13b_lora_llamacpp", allow_patterns=model_name, cache_dir="/data")
 
59
 
60
  rm_files = [os.path.join("/data", f) for f in os.listdir("/data")]
61
  for f in rm_files:
62
+ if os.path.isfile(f):
63
+ os.remove(f)
64
+ else:
65
+ shutil.rmtree(f)
66
 
67
  print("Downloading all files...")
68
  snapshot_download(repo_id=repo_name, local_dir="/data/saiga_13b_lora_llamacpp", allow_patterns=model_name, cache_dir="/data")