EmoCube commited on
Commit
7bc9207
·
verified ·
1 Parent(s): 17ae9a6

Update app.py

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