TDN-M commited on
Commit
35490f7
·
verified ·
1 Parent(s): ace0041

Update content_generation.py

Browse files
Files changed (1) hide show
  1. content_generation.py +2 -2
content_generation.py CHANGED
@@ -95,7 +95,7 @@ def create_content(prompt, content_type, language):
95
  try:
96
  client = Groq(api_key=GROQ_API_KEY)
97
  chat_completion = client.chat.completions.create(
98
- model="mixtral-8x7b-32768",
99
  messages=[
100
  {"role": "system", "content": general_instructions},
101
  {"role": "user", "content": prompt}
@@ -114,7 +114,7 @@ def extract_key_contents(script, num_contents=30):
114
  try:
115
  client = Groq(api_key=GROQ_API_KEY)
116
  chat_completion = client.chat.completions.create(
117
- model="mixtral-8x7b-32768",
118
  messages=[
119
  {"role": "system", "content": f"Bạn là một chuyên gia phân tích nội dung. Hãy trích xuất chính xác {num_contents} ý chính quan trọng nhất từ đoạn văn sau, mỗi ý không quá 20 từ."},
120
  {"role": "user", "content": script}
 
95
  try:
96
  client = Groq(api_key=GROQ_API_KEY)
97
  chat_completion = client.chat.completions.create(
98
+ model="llama3-8b-8192",
99
  messages=[
100
  {"role": "system", "content": general_instructions},
101
  {"role": "user", "content": prompt}
 
114
  try:
115
  client = Groq(api_key=GROQ_API_KEY)
116
  chat_completion = client.chat.completions.create(
117
+ model="llama3-8b-8192",
118
  messages=[
119
  {"role": "system", "content": f"Bạn là một chuyên gia phân tích nội dung. Hãy trích xuất chính xác {num_contents} ý chính quan trọng nhất từ đoạn văn sau, mỗi ý không quá 20 từ."},
120
  {"role": "user", "content": script}