huedaya commited on
Commit
6b10b2e
·
verified ·
1 Parent(s): f55fe3f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -19,7 +19,8 @@ app.add_middleware(
19
  allow_headers=["*"],
20
  )
21
 
22
- client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
 
23
 
24
  class Item(BaseModel):
25
  prompt: str
 
19
  allow_headers=["*"],
20
  )
21
 
22
+ # client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
23
+ client = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407")
24
 
25
  class Item(BaseModel):
26
  prompt: str