SalwaM commited on
Commit
fa39871
·
verified ·
1 Parent(s): b6f0207

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ if not api_key_coder:
14
 
15
  # --- 2. LLM CLIENT ---
16
  class GroqLLM:
17
- def __init__(self, api_key, model="GPT OSS 20B", temperature=0.0):
18
  self.client = Groq(api_key=api_key)
19
  self.model = model
20
  self.temperature = temperature
@@ -25,7 +25,7 @@ class GroqLLM:
25
  model=self.model,
26
  messages=[{"role": "user", "content": prompt}],
27
  temperature=self.temperature,
28
- max_tokens=4000
29
  )
30
  return response.choices[0].message.content
31
  except Exception as e:
 
14
 
15
  # --- 2. LLM CLIENT ---
16
  class GroqLLM:
17
+ def __init__(self, api_key, model="llama-3.3-70b-versatile", temperature=0.0):
18
  self.client = Groq(api_key=api_key)
19
  self.model = model
20
  self.temperature = temperature
 
25
  model=self.model,
26
  messages=[{"role": "user", "content": prompt}],
27
  temperature=self.temperature,
28
+ max_tokens=2000
29
  )
30
  return response.choices[0].message.content
31
  except Exception as e: