Saira20 commited on
Commit
852888b
·
verified ·
1 Parent(s): 517217d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -49,7 +49,10 @@ custom_role_conversions=None,
49
 
50
  # Import tool from Hub
51
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
52
- # hf_model_downloads = load_tool("m-ric/hf-model-downloads", trust_remote_code=True)
 
 
 
53
 
54
  with open("prompts.yaml", 'r') as stream:
55
  prompt_templates = yaml.safe_load(stream)
 
49
 
50
  # Import tool from Hub
51
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
52
+ model_download_tool = load_tool(
53
+ "m-ric/hf-model-downloads",
54
+ trust_remote_code=True
55
+ )
56
 
57
  with open("prompts.yaml", 'r') as stream:
58
  prompt_templates = yaml.safe_load(stream)