raghavNCI
commited on
Commit
·
606b2ff
1
Parent(s):
7edbb26
llm change
Browse files- question.py +1 -1
- requirements.txt +2 -1
question.py
CHANGED
@@ -16,7 +16,7 @@ HF_TOKEN = os.getenv("HF_TOKEN") # Hugging Face token for private models if nee
|
|
16 |
|
17 |
askMe = APIRouter()
|
18 |
|
19 |
-
client = InferenceClient("mistralai/Mistral-7B-Instruct", token=HF_TOKEN)
|
20 |
|
21 |
class QuestionInput(BaseModel):
|
22 |
question: str
|
|
|
16 |
|
17 |
askMe = APIRouter()
|
18 |
|
19 |
+
client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3", token=os.getenv("HF_TOKEN"))
|
20 |
|
21 |
class QuestionInput(BaseModel):
|
22 |
question: str
|
requirements.txt
CHANGED
@@ -5,4 +5,5 @@ python-dotenv
|
|
5 |
redis
|
6 |
transformers
|
7 |
torch
|
8 |
-
hf_xet
|
|
|
|
5 |
redis
|
6 |
transformers
|
7 |
torch
|
8 |
+
hf_xet
|
9 |
+
huggingface_hub
|