yxmnjxzx commited on
Commit
f77520e
·
verified ·
1 Parent(s): b98237e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,7 +66,7 @@ class PromptRefiner:
66
  def __init__(self, api_token: str):
67
  self.client = Groq(api_key=api_key)
68
 
69
- def generate_system_prompt(self, task_or_prompt: str, model: str = "llama-3.2-90b-text-preview") -> str:
70
  """Generate a system prompt for the given task or prompt."""
71
  messages = [
72
  {
@@ -82,7 +82,7 @@ class PromptRefiner:
82
  response = self.client.chat.completions.create(
83
  model=model,
84
  messages=messages,
85
- max_tokens=3000,
86
  temperature=0.7,
87
  stream=True
88
  )
 
66
  def __init__(self, api_token: str):
67
  self.client = Groq(api_key=api_key)
68
 
69
+ def generate_system_prompt(self, task_or_prompt: str, model: str = "llama-3.3-70b-versatile") -> str:
70
  """Generate a system prompt for the given task or prompt."""
71
  messages = [
72
  {
 
82
  response = self.client.chat.completions.create(
83
  model=model,
84
  messages=messages,
85
+ max_tokens=8000,
86
  temperature=0.7,
87
  stream=True
88
  )