Sarath0x8f commited on
Commit
e8fb4fe
1 Parent(s): dcf800f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -68,8 +68,8 @@ file_extractor = {
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
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