Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="
|
| 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=
|
| 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:
|