vsrinivas commited on
Commit
52acd0a
·
verified ·
1 Parent(s): 5e4d2e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -48,7 +48,8 @@ custom_role_conversions=None,
48
 
49
 
50
  # Import tool from Hub
51
- image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
52
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)
 
48
 
49
 
50
  # Import tool from Hub
51
+ # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
52
+ image_generation_tool = load_tool("vsrinivas/Image_Generation_by_SrinivasV", trust_remote_code=True)
53
 
54
  with open("prompts.yaml", 'r') as stream:
55
  prompt_templates = yaml.safe_load(stream)