Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def submit_message(prompt, prompt_template, good_foods, bad_foods, temperature,
|
|
62 |
answer = {'role': 'assistant', 'content': completion.choices[0].message.content }
|
63 |
history.append(answer)
|
64 |
|
65 |
-
state['total_tokens'] += completion
|
66 |
|
67 |
# completion2 = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=system_prompt + food_priming_prompt + history[-1:] + [classification_msg], temperature=temperature, max_tokens=max_tokens)
|
68 |
# print(completion2)
|
|
|
62 |
answer = {'role': 'assistant', 'content': completion.choices[0].message.content }
|
63 |
history.append(answer)
|
64 |
|
65 |
+
state['total_tokens'] += completion.usage.total_tokens
|
66 |
|
67 |
# completion2 = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=system_prompt + food_priming_prompt + history[-1:] + [classification_msg], temperature=temperature, max_tokens=max_tokens)
|
68 |
# print(completion2)
|