Spaces:
Running
Running
Sarath0x8f
commited on
Commit
•
e8fb4fe
1
Parent(s):
dcf800f
Update app.py
Browse files
app.py
CHANGED
@@ -68,8 +68,8 @@ file_extractor = {
|
|
68 |
}
|
69 |
|
70 |
# Embedding model and index initialization (to be populated by uploaded files)
|
71 |
-
|
72 |
-
embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-large-en") ## works good
|
73 |
# embed_model2 = HuggingFaceEmbedding(model_name="NeuML/pubmedbert-base-embeddings") ## works good
|
74 |
|
75 |
# sentence-transformers/distilbert-base-nli-mean-tokens
|
|
|
68 |
}
|
69 |
|
70 |
# Embedding model and index initialization (to be populated by uploaded files)
|
71 |
+
embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5") ## Works good and fast
|
72 |
+
# embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-large-en") ## works good
|
73 |
# embed_model2 = HuggingFaceEmbedding(model_name="NeuML/pubmedbert-base-embeddings") ## works good
|
74 |
|
75 |
# sentence-transformers/distilbert-base-nli-mean-tokens
|