IlyaGusev commited on
Commit
4d8b33d
1 Parent(s): 10b4ec0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -52,14 +52,15 @@ LOADER_MAPPING = {
52
  ".txt": (TextLoader, {"encoding": "utf8"}),
53
  }
54
 
 
55
  model_url = "https://huggingface.co/IlyaGusev/saiga2_13b_gguf/resolve/main/model-q4_K.gguf"
56
  repo_name = "IlyaGusev/saiga2_13b_gguf"
57
  model_name = "model-q4_K.gguf"
58
- final_model_path = f"/data/{model_name}"
59
  embedder_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
60
 
61
  print("Downloading all files...")
62
- rm_files = [os.path.join("/data", f) for f in os.listdir("/data")]
63
  for f in rm_files:
64
  if os.path.isfile(f):
65
  os.remove(f)
@@ -69,7 +70,7 @@ for f in rm_files:
69
  if not os.path.exists(final_model_path):
70
  with open(final_model_path, "wb") as f:
71
  http_get(model_url, f)
72
- os.chmod(final_model_path, 0o777)
73
  print("Files downloaded!")
74
 
75
  model = Llama(
 
52
  ".txt": (TextLoader, {"encoding": "utf8"}),
53
  }
54
 
55
+ directory = "."
56
  model_url = "https://huggingface.co/IlyaGusev/saiga2_13b_gguf/resolve/main/model-q4_K.gguf"
57
  repo_name = "IlyaGusev/saiga2_13b_gguf"
58
  model_name = "model-q4_K.gguf"
59
+ final_model_path = os.path.join(directory, model_name)
60
  embedder_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
61
 
62
  print("Downloading all files...")
63
+ rm_files = [os.path.join(directory, f) for f in os.listdir(directory)]
64
  for f in rm_files:
65
  if os.path.isfile(f):
66
  os.remove(f)
 
70
  if not os.path.exists(final_model_path):
71
  with open(final_model_path, "wb") as f:
72
  http_get(model_url, f)
73
+ os.chmod(final_model_path, 0o777)
74
  print("Files downloaded!")
75
 
76
  model = Llama(