ego commited on
Commit
e4170ab
·
1 Parent(s): a9d5e01

Revert LLM back to nvidia/nemotron-3-nano-30b-a3b, update UI description

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. core/models.py +1 -1
app.py CHANGED
@@ -228,7 +228,7 @@ def show_home():
228
  </div>
229
  <div style="border-left: 1px solid #eee;">
230
  <h4>🧬 Models</h4>
231
- <p style="font-size: 0.85rem; color: #666;">NVIDIA Gemma 4 31B (Reasoning)<br>NVIDIA Nemotron Embed-1B (Vector)<br>NVIDIA Riva TTS (Audio)</p>
232
  </div>
233
  </div>
234
  </div>
 
228
  </div>
229
  <div style="border-left: 1px solid #eee;">
230
  <h4>🧬 Models</h4>
231
+ <p style="font-size: 0.85rem; color: #666;">NVIDIA Nemotron-3 (Reasoning)<br>NVIDIA Nemotron Embed-1B (Vector)<br>NVIDIA Riva TTS (Audio)</p>
232
  </div>
233
  </div>
234
  </div>
core/models.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  import streamlit as st
3
  from langchain_nvidia_ai_endpoints import ChatNVIDIA, NVIDIAEmbeddings
4
 
5
- def get_llm(model_name: str = "google/gemma-4-31b-it"):
6
  api_key = os.getenv("NV_API_KEY")
7
  if not api_key and "NV_API_KEY" in st.secrets:
8
  api_key = st.secrets["NV_API_KEY"]
 
2
  import streamlit as st
3
  from langchain_nvidia_ai_endpoints import ChatNVIDIA, NVIDIAEmbeddings
4
 
5
+ def get_llm(model_name: str = "nvidia/nemotron-3-nano-30b-a3b"):
6
  api_key = os.getenv("NV_API_KEY")
7
  if not api_key and "NV_API_KEY" in st.secrets:
8
  api_key = st.secrets["NV_API_KEY"]