Final_Assignment_Template01

Sleeping

Karim0111 commited on Feb 4

Commit

555e7a6

verified ·

1 Parent(s): aab5144

Update agents.py

Files changed (1) hide show

agents.py CHANGED Viewed

@@ -22,7 +22,7 @@ with open("prompts.yaml", 'r') as stream:
 # WORKING MODEL SETUP
 react_model = LiteLLMModel(
-    model_id="gemini/gemini-1.5-flash",
     api_key=os.getenv("GEMINI_KEY"),
     temperature=0.2
 )

 # WORKING MODEL SETUP
 react_model = LiteLLMModel(
+    model_id="huggingface/gemini/gemini-1.5-flash",
     api_key=os.getenv("GEMINI_KEY"),
     temperature=0.2
 )