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}")
|