Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
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 ==
|
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
|