NavyDevilDoc commited on
Commit
e395dab
·
verified ·
1 Parent(s): 1d8bd27

Update src/app.py

Browse files
Files changed (1) hide show
  1. 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", "Llama 3.2 (Meta)": "llama3.2:latest", "Gemma 3 (Google)": "gemma3: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: