tejasri19 commited on
Commit
6f2a6cc
·
verified ·
1 Parent(s): 030c3b2

Update app.py

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