Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,6 @@ pp = pprint.PrettyPrinter(indent=4)
|
|
12 |
def generate_response(messages):
|
13 |
model_name = os.getenv("MODEL_NAME")
|
14 |
response = client.chat.completions.create(model=model_name, messages=messages, temperature=0.5, max_tokens=250)
|
15 |
-
spinner.stop()
|
16 |
print("Request:")
|
17 |
pp.pprint(messages)
|
18 |
print(f"Completion tokens: {response.usage.completion_tokens}, Prompt tokens: {response.usage.prompt_tokens}, Total tokens: {response.usage.total_tokens}")
|
|
|
12 |
def generate_response(messages):
|
13 |
model_name = os.getenv("MODEL_NAME")
|
14 |
response = client.chat.completions.create(model=model_name, messages=messages, temperature=0.5, max_tokens=250)
|
|
|
15 |
print("Request:")
|
16 |
pp.pprint(messages)
|
17 |
print(f"Completion tokens: {response.usage.completion_tokens}, Prompt tokens: {response.usage.prompt_tokens}, Total tokens: {response.usage.total_tokens}")
|