Hesamnasiri commited on
Commit
52ea860
·
verified ·
1 Parent(s): 449a488

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -27,6 +27,8 @@ RERANK_MODEL_NAME = "BAAI/bge-reranker-large" # cross-encoder reranker
27
  COLLECTION_NAME = "pdf_rag_en"
28
 
29
  DEFAULT_MODELS = [
 
 
30
  "Qwen/Qwen2.5-7B-Instruct",
31
  "meta-llama/Meta-Llama-3.1-8B-Instruct",
32
  "mistralai/Mistral-7B-Instruct-v0.3",
 
27
  COLLECTION_NAME = "pdf_rag_en"
28
 
29
  DEFAULT_MODELS = [
30
+ "Qwen/Qwen2.5-1.5B-Instruct", # CPU-friendly
31
+ "microsoft/Phi-3.1-mini-4k-instruct", # CPU-friendly
32
  "Qwen/Qwen2.5-7B-Instruct",
33
  "meta-llama/Meta-Llama-3.1-8B-Instruct",
34
  "mistralai/Mistral-7B-Instruct-v0.3",