rishabh5752 commited on
Commit
a3c89e6
·
verified ·
1 Parent(s): 574373f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ def extract_text_from_pdf(pdf):
20
  def main():
21
  llm = LlamaCPP(
22
  model_url=None,
23
- model_path='models/phi-2.Q4_K_M.gguf',
24
  temperature=0.1,
25
  max_new_tokens=512,
26
  context_window=2048,
@@ -31,7 +31,7 @@ def main():
31
  )
32
 
33
  embed_model = LangchainEmbedding(
34
- HuggingFaceEmbeddings(model_name="BAAI/bge-small-en-v1.5")
35
  )
36
 
37
  service_context = ServiceContext.from_defaults(
 
20
  def main():
21
  llm = LlamaCPP(
22
  model_url=None,
23
+ model_path='phi-2.Q4_K_M.gguf',
24
  temperature=0.1,
25
  max_new_tokens=512,
26
  context_window=2048,
 
31
  )
32
 
33
  embed_model = LangchainEmbedding(
34
+ HuggingFaceEmbeddings(model_name="bge-small-en-v1.5")
35
  )
36
 
37
  service_context = ServiceContext.from_defaults(