Spaces:
Running
Running
Update rag_engine.py
Browse files- 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 |
-
|
71 |
except KeyError:
|
72 |
-
print("❌ Error: Embedding model
|
73 |
return None, None
|
74 |
|
75 |
# Load tokenizer and model
|
76 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
77 |
model = AutoModel.from_pretrained(
|
78 |
-
|
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 |
|