Javier Vera commited on
Commit
f45ad41
·
verified ·
1 Parent(s): 9e91c10

Update rag_hf.py

Browse files
Files changed (1) hide show
  1. rag_hf.py +2 -1
rag_hf.py CHANGED
@@ -14,7 +14,8 @@ from dotenv import load_dotenv
14
  # === CONFIGURATION ===
15
  load_dotenv()
16
 
17
- ENDPOINT_URL = os.getenv("HF_ENDPOINT")
 
18
  HF_API_TOKEN = os.getenv("HF_API_TOKEN")
19
  EMBEDDING_MODEL = "intfloat/multilingual-e5-base"
20
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
14
  # === CONFIGURATION ===
15
  load_dotenv()
16
 
17
+ ENDPOINT_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
18
+
19
  HF_API_TOKEN = os.getenv("HF_API_TOKEN")
20
  EMBEDDING_MODEL = "intfloat/multilingual-e5-base"
21
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"