venthan commited on
Commit
fefff88
1 Parent(s): 17a9c23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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['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)
 
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)