ianeksdi commited on
Commit
feba4a3
·
verified ·
1 Parent(s): 01e6c19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -3,13 +3,13 @@ import yaml
3
  from tools.final_answer import FinalAnswerTool
4
  from Gradio_UI import GradioUI
5
 
6
- # Define the system prompt for hypertension prevention advice.
7
  system_prompt = (
8
  "You are a health and lifestyle advisor specializing in the early detection and prevention of hypertension. "
9
- "The diagnostic criteria are as follows: Normal BP is < 120/80 mmHg, Borderline BP is 120-139/80-89 mmHg, "
10
- "and Hypertension is > 140/90 mmHg. Provide direct, concise, and actionable advice based solely on the user's details. "
11
- "Do NOT include any internal reasoning, chain-of-thought, or meta commentary in your final response. "
12
- "Only output the final advice. For example, if the user mentions alcohol consumption, simply say: "
13
  "'Reduce alcohol intake, as it can raise blood pressure.'"
14
  )
15
 
@@ -20,18 +20,18 @@ final_answer = FinalAnswerTool()
20
  model = HfApiModel(
21
  max_tokens=2096,
22
  temperature=0.5,
23
- model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B', # This model might be overloaded; consider alternatives if needed.
24
  custom_role_conversions=None,
25
  )
26
 
27
- # Load prompt templates from a YAML file.
28
  with open("prompts.yaml", 'r') as stream:
29
  prompt_templates = yaml.safe_load(stream)
30
 
31
- # Initialize the CodeAgent with the final_answer tool and your system prompt as its description.
32
  agent = CodeAgent(
33
  model=model,
34
- tools=[final_answer], # Only keep the final answer tool.
35
  max_steps=6,
36
  verbosity_level=1,
37
  grammar=None,
 
3
  from tools.final_answer import FinalAnswerTool
4
  from Gradio_UI import GradioUI
5
 
6
+ # Updated system prompt: provide only final, direct advice with no chain-of-thought or code.
7
  system_prompt = (
8
  "You are a health and lifestyle advisor specializing in the early detection and prevention of hypertension. "
9
+ "Diagnostic criteria: Normal BP is < 120/80 mmHg, Borderline BP is 120-139/80-89 mmHg, and Hypertension is > 140/90 mmHg. "
10
+ "Based solely on the user's details, provide only the final, direct, and concise lifestyle tips. "
11
+ "Do NOT include any internal reasoning, chain-of-thought, or any code snippets in your output. "
12
+ "Only output the final advice as plain text. For example, if the user mentions alcohol consumption, simply say: "
13
  "'Reduce alcohol intake, as it can raise blood pressure.'"
14
  )
15
 
 
20
  model = HfApiModel(
21
  max_tokens=2096,
22
  temperature=0.5,
23
+ model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B', # Adjust if needed.
24
  custom_role_conversions=None,
25
  )
26
 
27
+ # Load prompt templates from the YAML file.
28
  with open("prompts.yaml", 'r') as stream:
29
  prompt_templates = yaml.safe_load(stream)
30
 
31
+ # Initialize the CodeAgent with the updated system prompt.
32
  agent = CodeAgent(
33
  model=model,
34
+ tools=[final_answer],
35
  max_steps=6,
36
  verbosity_level=1,
37
  grammar=None,