Update app.py
Browse files
app.py
CHANGED
|
@@ -64,10 +64,8 @@ GROQ_MODELS = [
|
|
| 64 |
]
|
| 65 |
|
| 66 |
OPENROUTER_MODELS = [
|
| 67 |
-
"nvidia/nemotron-3
|
| 68 |
-
"
|
| 69 |
-
"microsoft/mai-transcribe-1.5",
|
| 70 |
-
"google/gemini-embedding-2"
|
| 71 |
]
|
| 72 |
|
| 73 |
# --- Reactive State Event Handlers ---
|
|
@@ -103,12 +101,10 @@ def call_llm(provider, api_key, model_choice, system_prompt, history_messages):
|
|
| 103 |
elif provider == "OpenRouter":
|
| 104 |
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
|
| 105 |
openrouter_model_map = {
|
| 106 |
-
"nvidia/nemotron-3
|
| 107 |
-
"
|
| 108 |
-
"microsoft/mai-transcribe-1.5": "microsoft/mai-transcribe-1.5",
|
| 109 |
-
"google/gemini-embedding-2": "google/gemini-embedding-2"
|
| 110 |
}
|
| 111 |
-
target_model = openrouter_model_map.get(model_choice, "nvidia/nemotron-3
|
| 112 |
|
| 113 |
payload = {
|
| 114 |
"model": target_model,
|
|
|
|
| 64 |
]
|
| 65 |
|
| 66 |
OPENROUTER_MODELS = [
|
| 67 |
+
"nvidia/nemotron-3-ultra-550b-a55b:free",
|
| 68 |
+
"nvidia/nemotron-3.5-content-safety:free"
|
|
|
|
|
|
|
| 69 |
]
|
| 70 |
|
| 71 |
# --- Reactive State Event Handlers ---
|
|
|
|
| 101 |
elif provider == "OpenRouter":
|
| 102 |
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
|
| 103 |
openrouter_model_map = {
|
| 104 |
+
"nvidia/nemotron-3-ultra-550b-a55b:free": "nvidia/nemotron-3-ultra-550b-a55b:free",
|
| 105 |
+
"nvidia/nemotron-3.5-content-safety:free": "nvidia/nemotron-3.5-content-safety:free"
|
|
|
|
|
|
|
| 106 |
}
|
| 107 |
+
target_model = openrouter_model_map.get(model_choice, "nvidia/nemotron-3-ultra-550b-a55b:free")
|
| 108 |
|
| 109 |
payload = {
|
| 110 |
"model": target_model,
|