Update app.py
Browse files
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 =
|
59 |
embedder_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
|
60 |
|
61 |
print("Downloading all files...")
|
62 |
-
rm_files = [os.path.join(
|
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 |
-
|
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(
|