Chittrarasu commited on
Commit
ea5aa8c
·
1 Parent(s): 84cf206
Files changed (2) hide show
  1. .gitignore +1 -0
  2. models/model_loader.py +1 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ /tmp/sentence_transformer/
models/model_loader.py CHANGED
@@ -2,7 +2,7 @@ from sentence_transformers import SentenceTransformer
2
  import os
3
 
4
  # Model path and Hugging Face model ID
5
- model_dir = 'models/sentence_transformer'
6
  hf_model_id = 'Alibaba-NLP/gte-base-en-v1.5'
7
 
8
  # Check if model directory exists
 
2
  import os
3
 
4
  # Model path and Hugging Face model ID
5
+ model_dir = '/tmp/sentence_transformer' # Change here
6
  hf_model_id = 'Alibaba-NLP/gte-base-en-v1.5'
7
 
8
  # Check if model directory exists