Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -11,13 +11,13 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
|
|
11 |
from huggingface_hub import hf_hub_download
|
12 |
from llama_index.core.node_parser import SentenceSplitter
|
13 |
|
14 |
-
# Fungsi untuk mengunduh model
|
15 |
def initialize_llama_model():
|
16 |
# Unduh model jika belum ada di direktori kerja
|
17 |
model_path = hf_hub_download(
|
18 |
-
repo_id="
|
19 |
-
filename="
|
20 |
-
cache_dir="./models" # Lokasi direktori
|
21 |
)
|
22 |
return model_path
|
23 |
|
|
|
11 |
from huggingface_hub import hf_hub_download
|
12 |
from llama_index.core.node_parser import SentenceSplitter
|
13 |
|
14 |
+
# Fungsi untuk mengunduh model Komodo-7B
|
15 |
def initialize_llama_model():
|
16 |
# Unduh model jika belum ada di direktori kerja
|
17 |
model_path = hf_hub_download(
|
18 |
+
repo_id="Yellow-AI-NLP/komodo-7b-base", # Repo model di Hugging Face
|
19 |
+
filename="komodo-7b-base.Q4_K_M.gguf", # Nama file model (pastikan tersedia dalam format GGUF)
|
20 |
+
cache_dir="./models" # Lokasi direktori penyimpanan model
|
21 |
)
|
22 |
return model_path
|
23 |
|