Update app.py

app.py

@@ -47,12 +47,14 @@ PROMPT_STRATEGIES = {
 }
 
 def create_assistant(model_name):
-
     client = OpenAI(
         base_url = "https://integrate.api.nvidia.com/v1",
         api_key = API_KEY
     )
     
+    # Should use MODELS dictionary to get the actual model name
+    model_name = MODELS[model_name]["name"]  # Add this line
+    
     return AIAssistant(
         client=client,
         model=model_name