Update app.py

app.py

@@ -25,14 +25,30 @@ def DuckDuckGoSearchTool(query: str) -> str:
 
 final_answer = FinalAnswerTool()
 
-model = HfApiModel(
-    max_tokens=10000,
-    temperature=1,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
-)
+# Define both model IDs
+primary_model_id = 'Qwen/Qwen2.5-Coder-32B-Instruct'
+fallback_model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
+
+# Try to initialize the primary model
+try:
+    model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id=primary_model_id,
+        custom_role_conversions=None,
+    )
+    print(f"Using primary model: {primary_model_id}")
+except Exception as e:
+    print(f"Error initializing primary model: {e}")
+    print(f"Falling back to secondary model: {fallback_model_id}")
+    # If the primary model fails, use the fallback model
+    model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id=fallback_model_id,
+        custom_role_conversions=None,
+    )
 
-# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 # Загрузка prompt_templates из prompts.yaml
@@ -46,14 +62,7 @@ with open("prompts.yaml", 'r') as stream:
 if isinstance(prompt_templates, dict) and 'system_prompt' in prompt_templates:
     # Если все ок, используем загруженный шаблон
     system_prompt = prompt_templates['system_prompt']
-else:
-    # Иначе, создаем новый словарь с system_prompt
-    system_prompt = """
-    Ты — Perplexity: профессиональный поисковый ассистент.
-    Твоя задача — предоставлять исчерпывающие и точные ответы на запросы пользователей, основываясь на информации из нескольких источников в интернете.
-    #... (остальной текст вашего промпта)...
-    Начни!
-    """
+
     prompt_templates = {'system_prompt': system_prompt}
 
 agent = CodeAgent(