ankanghosh commited on
Commit
72511f1
·
verified ·
1 Parent(s): ef4e95d

Update rag_engine.py

Browse files
Files changed (1) hide show
  1. rag_engine.py +4 -4
rag_engine.py CHANGED
@@ -67,15 +67,15 @@ def load_model():
67
 
68
  # Get embedding model path from secrets
69
  try:
70
- embedding_model_path = st.secrets["embedding_model_path"]
71
  except KeyError:
72
- print("❌ Error: Embedding model path not found in secrets")
73
  return None, None
74
 
75
  # Load tokenizer and model
76
- tokenizer = AutoTokenizer.from_pretrained(embedding_model_path)
77
  model = AutoModel.from_pretrained(
78
- embedding_model_path,
79
  torch_dtype=torch.float16
80
  )
81
 
 
67
 
68
  # Get embedding model path from secrets
69
  try:
70
+ embedding_model = st.secrets["embedding_model"]
71
  except KeyError:
72
+ print("❌ Error: Embedding model not found in secrets")
73
  return None, None
74
 
75
  # Load tokenizer and model
76
+ tokenizer = AutoTokenizer.from_pretrained(embedding_model)
77
  model = AutoModel.from_pretrained(
78
+ embedding_model,
79
  torch_dtype=torch.float16
80
  )
81