Update app.py

app.py

@@ -6,13 +6,38 @@ import pandas as pd
 from smolagents import HfApiModel, FinalAnswerTool
 from smolagents import DuckDuckGoSearchTool, LiteLLMModel, PythonInterpreterTool, CodeAgent, SpeechToTextTool, WikipediaSearchTool, VisitWebpageTool
 
+from litellm import Router
+
+model_list = [
+    {
+        "model_name": "deepseek-r1-0528-key1",
+        "litellm_params": {
+            "model_id": "openrouter/deepseek/deepseek-r1-0528:free",
+            "api_key": "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",
+            "max_retries": 3
+        }
+    },
+    {
+        "model_name": "deepseek-r1-0528-key2",
+        "litellm_params": {
+            "model_id": "openrouter/deepseek/deepseek-r1-0528:free",
+            "api_key": "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",
+            "max_retries": 3
+        }
+    },
+    # Add more keys/deployments as needed
+]
+
+
+
 #model = HfApiModel(
 #    model_id="Qwen/Qwen3-235B-A22B",
 #    api_key=os.getenv("HFT")  # Ensure this environment variable is set
 #)
-model = LiteLLMModel(model_id="openrouter/deepseek/deepseek-r1-0528:free", api_key="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",max_retries=3)
+#model = LiteLLMModel(model_id="openrouter/deepseek/deepseek-r1-0528:free", api_key="",max_retries=3)
 
 
+model = Router(model_list=model_list)