raghavNCI commited on
Commit
606b2ff
·
1 Parent(s): 7edbb26

llm change

Browse files
Files changed (2) hide show
  1. question.py +1 -1
  2. 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