Update app.py

app.py

@@ -12,15 +12,15 @@ from tools import evaluate_consumption, evaluate_consumption_example
 from retriever import FrugalAI_methods
 
 #HF_TOKEN = os.environ.get("HF_TOKEN") 
-#try:
- #   API_KEY = os.environ.get("API_KEY") 
-#except:
- #   print("Please provide the Anthropic API_KEY in the secret variables")
+try:
+    API_KEY = os.environ.get("API_KEY") 
+except:
+    print("Please provide the Anthropic API_KEY in the secret variables")
 # Initialize the Hugging Face model
 model = LiteLLMModel(
-    model_id="anthropic/claude-3-5-sonnet-latest",
+    model_id="gpt-4",
     temperature=0.2,
-   # api_key=API_KEY
+    api_key=API_KEY
 )
 
 evaluate_consumption = evaluate_consumption()