mitulagr2 commited on
Commit
d516513
1 Parent(s): 95d75bf

Update rag.py

Browse files
Files changed (1) hide show
  1. app/rag.py +3 -2
app/rag.py CHANGED
@@ -43,7 +43,8 @@ class ChatPDF:
43
  )
44
 
45
  llm = LlamaCPP(
46
- model_url="https://huggingface.co/Qwen/Qwen1.5-1.8B-Chat-GGUF/resolve/main/qwen1_5-1_8b-chat-q4_k_m.gguf",
 
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=6,
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):