demoPOC commited on
Commit
3222362
·
1 Parent(s): 21ab5a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -59,7 +59,7 @@ uploads_dir = os.path.join(app.root_path,'static', 'uploads')
59
 
60
  os.makedirs(uploads_dir, exist_ok=True)
61
 
62
- defaultEmbeddingModelID = 0
63
  defaultLLMID=0
64
 
65
 
@@ -77,7 +77,7 @@ def getEmbeddingModel(embeddingId):
77
  model_kwargs = {'device': 'cpu'}
78
  encode_kwargs = {'normalize_embeddings': True}
79
  embeddings = HuggingFaceInstructEmbeddings(model_name=model_name,model_kwargs=model_kwargs,encode_kwargs=encode_kwargs)
80
- elif (embeddingId == 2):
81
  model_name = "BAAI/bge-large-en-v1.5"
82
  model_kwargs = {'device': 'cuda'}
83
  encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity
 
59
 
60
  os.makedirs(uploads_dir, exist_ok=True)
61
 
62
+ defaultEmbeddingModelID = 3
63
  defaultLLMID=0
64
 
65
 
 
77
  model_kwargs = {'device': 'cpu'}
78
  encode_kwargs = {'normalize_embeddings': True}
79
  embeddings = HuggingFaceInstructEmbeddings(model_name=model_name,model_kwargs=model_kwargs,encode_kwargs=encode_kwargs)
80
+ elif (embeddingId == 3):
81
  model_name = "BAAI/bge-large-en-v1.5"
82
  model_kwargs = {'device': 'cuda'}
83
  encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity