Update app.py

app.py

@@ -22,25 +22,11 @@ model = LiteLLMModel(
     temperature=0.2,
     api_key=os.environ["API_KEY"]
 )
-#model = InferenceClientModel("deepseek-ai/DeepSeek-R1",max_tokens=500, token=HF_TOKEN)
-#headers = {
- #   "Authorization": f"Bearer {HF_TOKEN}"
-#}
 
-# Initialize the weather tool
 evaluate_consumption = evaluate_consumption()
 evaluate_consumption_example = evaluate_consumption_example()
 FrugalAI_methods=FrugalAI_methods()
 
-modified_prompt = (
-    "\nYou are an AI agent that receives Python code from a manager. "
-    "Your task is to test its resource consumption using a custom tool. "
-    "Then, propose frugal alternatives, such as pruning, using your dedicated tool — always use it. "
-    "Additionally, feel free to search the internet for frugal methods and present them to the manager. "
-    "Examples include Knowledge Distillation, Transfer Learning, and others."
-)
-
-
 # Create agent with all the tools
 fruggy = CodeAgent(
     tools=[evaluate_consumption, evaluate_consumption_example, FrugalAI_methods],