Spaces:
Sleeping
Sleeping
Update src/app.py
Browse files- src/app.py +6 -1
src/app.py
CHANGED
|
@@ -318,7 +318,9 @@ with st.sidebar:
|
|
| 318 |
|
| 319 |
st.subheader("2. Chat Model")
|
| 320 |
# Base local models
|
| 321 |
-
model_map = {"Granite 4 (IBM)": "granite4:latest",
|
|
|
|
|
|
|
| 322 |
opts = list(model_map.keys())
|
| 323 |
|
| 324 |
is_admin = "admin" in st.session_state.roles
|
|
@@ -333,6 +335,9 @@ with st.sidebar:
|
|
| 333 |
# Add Premium Options if Admin or Key provided
|
| 334 |
if is_admin or st.session_state.get("user_openai_key"):
|
| 335 |
opts.append("GPT-4o (Omni)")
|
|
|
|
|
|
|
|
|
|
| 336 |
|
| 337 |
# Add Gemini if Key exists (System wide)
|
| 338 |
if GOOGLE_KEY:
|
|
|
|
| 318 |
|
| 319 |
st.subheader("2. Chat Model")
|
| 320 |
# Base local models
|
| 321 |
+
model_map = {"Granite 4 (IBM)": "granite4:latest",
|
| 322 |
+
"Llama 3.2 (Meta)": "llama3.2:latest",
|
| 323 |
+
"Gemma 3 (Google)": "gemma3:latest"}
|
| 324 |
opts = list(model_map.keys())
|
| 325 |
|
| 326 |
is_admin = "admin" in st.session_state.roles
|
|
|
|
| 335 |
# Add Premium Options if Admin or Key provided
|
| 336 |
if is_admin or st.session_state.get("user_openai_key"):
|
| 337 |
opts.append("GPT-4o (Omni)")
|
| 338 |
+
|
| 339 |
+
if is_admin: # Only admins should see this initially to save RAM
|
| 340 |
+
opts.append("Custom Gemma 2 9B (Fine-Tuned)")
|
| 341 |
|
| 342 |
# Add Gemini if Key exists (System wide)
|
| 343 |
if GOOGLE_KEY:
|