EmoCube commited on
Commit
c867a85
·
verified ·
1 Parent(s): 590959c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def get_messages_api(messages):
53
  return json.dumps(messages, indent=4)
54
 
55
  def run_huggingface_model(model, messages, max_tokens, temperature, top_p):
56
- client = InferenceClient(model)
57
  response = client.chat_completion(
58
  messages,
59
  max_tokens=max_tokens,
 
53
  return json.dumps(messages, indent=4)
54
 
55
  def run_huggingface_model(model, messages, max_tokens, temperature, top_p):
56
+ client = InferenceClient(model, token = os.environ.get("HF_TOKEN"))
57
  response = client.chat_completion(
58
  messages,
59
  max_tokens=max_tokens,