Zaious commited on
Commit
ac6a145
·
verified ·
1 Parent(s): 7818f62

Update ideaGen.py

Browse files
Files changed (1) hide show
  1. ideaGen.py +2 -2
ideaGen.py CHANGED
@@ -60,7 +60,7 @@ def ideagen(content_type, brand_name, product_name, target_audience, core_essenc
60
 
61
  #for _ in range(loop):
62
  response = client.chat.completions.create(
63
- model='gpt-4-turbo',
64
  max_tokens=2000,
65
  temperature=0.7,
66
  messages=messages_base
@@ -68,7 +68,7 @@ def ideagen(content_type, brand_name, product_name, target_audience, core_essenc
68
  completed_text = response.choices[0].message.content
69
  total_prompt_tokens_used += response.usage.prompt_tokens
70
  total_completion_tokens_used += response.usage.completion_tokens
71
- price = total_prompt_tokens_used*0.03/1000 + total_completion_tokens_used*0.06/1000
72
 
73
  full_text += completed_text + "\n\n----------\n\n"
74
  total_price += price
 
60
 
61
  #for _ in range(loop):
62
  response = client.chat.completions.create(
63
+ model='gpt-4o',
64
  max_tokens=2000,
65
  temperature=0.7,
66
  messages=messages_base
 
68
  completed_text = response.choices[0].message.content
69
  total_prompt_tokens_used += response.usage.prompt_tokens
70
  total_completion_tokens_used += response.usage.completion_tokens
71
+ price = total_prompt_tokens_used*0.005/1000 + total_completion_tokens_used*0.015/1000
72
 
73
  full_text += completed_text + "\n\n----------\n\n"
74
  total_price += price