We leverage it to improve our prompt | |
a rabbit wearing a space suit: | |
thon | |
from transformers import HfAgent | |
agent = HfAgent("https://api-inference.huggingface.co/models/bigcode/starcoder", additional_tools=[tool]) | |
agent.run("Generate an image of the prompt after improving it. |