sksameermujahid commited on
Commit
eae720b
·
verified ·
1 Parent(s): 54eb4be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -60,8 +60,14 @@ conversation_context = {}
60
  def load_sentence_transformer():
61
  print("Loading SentenceTransformer model...")
62
  try:
 
 
 
63
  from sentence_transformers import SentenceTransformer
64
- model_embedding = SentenceTransformer('jinaai/jina-embeddings-v2-base-en').to(device)
 
 
 
65
 
66
  if not os.path.exists(MODEL_PATH):
67
  raise FileNotFoundError(f"Model state dict not found at {MODEL_PATH}. Please ensure the file exists.")
 
60
  def load_sentence_transformer():
61
  print("Loading SentenceTransformer model...")
62
  try:
63
+ # Set environment variable to auto-accept trust_remote_code
64
+ os.environ['TRUST_REMOTE_CODE'] = "1"
65
+
66
  from sentence_transformers import SentenceTransformer
67
+ model_embedding = SentenceTransformer(
68
+ 'jinaai/jina-embeddings-v3',
69
+ trust_remote_code=True
70
+ ).to(device)
71
 
72
  if not os.path.exists(MODEL_PATH):
73
  raise FileNotFoundError(f"Model state dict not found at {MODEL_PATH}. Please ensure the file exists.")