Spaces:

atharva-nlp
/

hg_agent

Sleeping

atharva-nlp commited on Dec 9, 2024

Commit

27ecc8f

verified ·

1 Parent(s): 361f161

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from transformers import (
 # Import tool from Hub
 image_generation_tool = load_tool("m-ric/text-to-image")
-llm_engine = HfApiEngine("meta-llama/Meta-Llama-3-70B-Instruct")
 # Initialize the agent with the image generation tool
 agent = ReactCodeAgent(tools=[image_generation_tool], llm_engine=llm_engine)

 # Import tool from Hub
 image_generation_tool = load_tool("m-ric/text-to-image")
+llm_engine = HfApiEngine("HuggingFaceH4/zephyr-7b-beta")
 # Initialize the agent with the image generation tool
 agent = ReactCodeAgent(tools=[image_generation_tool], llm_engine=llm_engine)