ManuelMagana commited on
Commit
a24abd8
·
1 Parent(s): 8f5f876

Adding variable to access to the private model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from huggingface_hub import InferenceClient
8
  """
9
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
10
  """
11
- client = InferenceClient(model="Layer7/autotrain-llama")
12
 
13
 
14
  def respond(message, history: list[tuple[str, str]], system_message, max_tokens, temperature, top_p):
 
8
  """
9
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
10
  """
11
+ client = InferenceClient(model="Layer7/autotrain-llama", token=os.getenv("HF_TOKEN"))
12
 
13
 
14
  def respond(message, history: list[tuple[str, str]], system_message, max_tokens, temperature, top_p):