tejasri19 commited on
Commit
7a6c9a4
·
verified ·
1 Parent(s): 9c7bb3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -78,6 +78,8 @@ model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(
78
 
79
  # Import tool from Hub
80
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 
81
 
82
  with open("prompts.yaml", 'r') as stream:
83
  prompt_templates = yaml.safe_load(stream)
 
78
 
79
  # Import tool from Hub
80
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
81
+ result = image_generation_tool("A dog sitting on a chair")
82
+ print(result)
83
 
84
  with open("prompts.yaml", 'r') as stream:
85
  prompt_templates = yaml.safe_load(stream)