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