Update app.py

app.py

@@ -138,7 +138,7 @@ print(get_stock_data_baostock('sh.600941'))
 final_answer = FinalAnswerTool()
 
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True, token=os.getenv('hf_token'))
+image_generation_tool = load_tool("m-ric/text-to-image", trust_remote_code=True, token=os.getenv('hf_token'))
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
@@ -146,7 +146,11 @@ with open("prompts.yaml", 'r') as stream:
 model = HfApiModel(
     max_tokens=16000,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    #model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
+    #model_id='mistralai/Mistral-7B-Instruct-v0.3',
+    #model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',
+    #model_id="google/gemma-2-9b-it",
     custom_role_conversions=None,
     token=os.getenv('hf_token'),
 )