Marcos12886 commited on
Commit
5b24797
1 Parent(s): 214fa62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,9 +13,9 @@ pipe = pipeline(
13
  feature_extractor=FEATURE_EXTRACTOR,
14
  device="cuda",
15
  )
16
- access_token = os.getenv('HF_TOKEN')
17
- # client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct", token=access_token)
18
- client = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407", token=access_token)
19
 
20
  def respond(
21
  message,
 
13
  feature_extractor=FEATURE_EXTRACTOR,
14
  device="cuda",
15
  )
16
+ token = os.getenv('HF_TOKEN')
17
+ # client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct", token=token)
18
+ client = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407", token=token)
19
 
20
  def respond(
21
  message,