Spaces:
Running
Running
Update app.py
Browse files
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,
|