Update app.py

app.py

@@ -74,12 +74,12 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
                 "Additionally, feel free to search the internet for frugal methods and present them to the manager. "
                 "Examples include Knowledge Distillation, Transfer Learning, and others. If implementation guidance is given by your custom tools, always give it in your final answer to the manager. "
                 "Here is my code: "
-                "from litellm import completion"
+                "import anthropic"
                 "API_KEY = os.environ.get('API_KEY') "
-                "model = LiteLLMModel(model_id='anthropic/claude-3-5-sonnet-latest',temperature=0.2,api_key=os.environ['API_KEY'])"
-                "messages = [{"role": "user", "content": "Hey! how's it going?"}],"
-                "response = completion(model=model, messages=messages)"
-                "print(response). Please give me frugal alternatives."
+                "client = anthropic.Anthropic(API_KEY)"
+                "message = client.messages.create(model='claude-sonnet-4-20250514',max_tokens=1024,messages=[{"role": "user", "content": "Hello, Claude"}])"
+                "print(message.content)"
+                "Please give me frugal alternatives."
             ],
                 description="Share your Python code with this AI agent! It will track its CO2 emissions using CodeCarbon and recommend greener, frugal AI alternatives."
             )