Spaces:
Runtime error
Runtime error
Update rag.py
Browse files- app/rag.py +3 -2
app/rag.py
CHANGED
@@ -43,7 +43,8 @@ class ChatPDF:
|
|
43 |
)
|
44 |
|
45 |
llm = LlamaCPP(
|
46 |
-
|
|
|
47 |
temperature=0.1,
|
48 |
max_new_tokens=256,
|
49 |
context_window=3900, #32k
|
@@ -103,7 +104,7 @@ class ChatPDF:
|
|
103 |
|
104 |
self.query_engine = index.as_query_engine(
|
105 |
streaming=True,
|
106 |
-
similarity_top_k=
|
107 |
)
|
108 |
|
109 |
def ask(self, query: str):
|
|
|
43 |
)
|
44 |
|
45 |
llm = LlamaCPP(
|
46 |
+
# https://huggingface.co/MaziyarPanahi/Qwen2-1.5B-Instruct-GGUF/resolve/main/Qwen2-1.5B-Instruct.Q4_K_M.gguf
|
47 |
+
model_url="https://huggingface.co/Qwen/Qwen2-0.5B-Instruct-GGUF/resolve/main/qwen2-0_5b-instruct-fp16.gguf",
|
48 |
temperature=0.1,
|
49 |
max_new_tokens=256,
|
50 |
context_window=3900, #32k
|
|
|
104 |
|
105 |
self.query_engine = index.as_query_engine(
|
106 |
streaming=True,
|
107 |
+
similarity_top_k=3,
|
108 |
)
|
109 |
|
110 |
def ask(self, query: str):
|