Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ from sentence_transformers.util import cos_sim
|
|
25 |
from llama_cpp import Llama
|
26 |
|
27 |
|
28 |
-
SYSTEM_PROMPT = "
|
29 |
|
30 |
LOADER_MAPPING = {
|
31 |
".csv": (CSVLoader, {}),
|
@@ -45,8 +45,8 @@ LOADER_MAPPING = {
|
|
45 |
|
46 |
def load_model(
|
47 |
directory: str = ".",
|
48 |
-
model_name: str = "model-
|
49 |
-
model_url: str = "https://huggingface.co/
|
50 |
):
|
51 |
final_model_path = os.path.join(directory, model_name)
|
52 |
|
|
|
25 |
from llama_cpp import Llama
|
26 |
|
27 |
|
28 |
+
SYSTEM_PROMPT = "Welcome to the CNCF Bot"
|
29 |
|
30 |
LOADER_MAPPING = {
|
31 |
".csv": (CSVLoader, {}),
|
|
|
45 |
|
46 |
def load_model(
|
47 |
directory: str = ".",
|
48 |
+
model_name: str = "ggml-model-Q4_K_M.gguf",
|
49 |
+
model_url: str = "https://huggingface.co/julioc-p/CNCF/blob/main/ggml-model-Q4_K_M.gguf"
|
50 |
):
|
51 |
final_model_path = os.path.join(directory, model_name)
|
52 |
|