Andrey commited on
Commit
cbb36a7
·
verified ·
1 Parent(s): dd382f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -39,20 +39,20 @@ model = HfApiModel(
39
  max_tokens=2096,
40
  temperature=0.5,
41
  # model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
42
- # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
43
- model_id='https://jc26mwg228mkj8dw.us-east-1.aws.endpoints.huggingface.cloud',
44
  custom_role_conversions=None,
45
  )
46
 
47
 
48
  # Import tool from Hub
49
- # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
50
 
51
- image_generation_tool = Tool.from_space(
52
- "black-forest-labs/FLUX.1-schnell",
53
- name="image_generator",
54
- description="Generate an image from a prompt"
55
- )
56
 
57
 
58
  with open("prompts.yaml", 'r') as stream:
 
39
  max_tokens=2096,
40
  temperature=0.5,
41
  # model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
42
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
43
+ # model_id='https://jc26mwg228mkj8dw.us-east-1.aws.endpoints.huggingface.cloud',
44
  custom_role_conversions=None,
45
  )
46
 
47
 
48
  # Import tool from Hub
49
+ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
50
 
51
+ # image_generation_tool = Tool.from_space(
52
+ # "black-forest-labs/FLUX.1-schnell",
53
+ # name="image_generator",
54
+ # description="Generate an image from a prompt"
55
+ # )
56
 
57
 
58
  with open("prompts.yaml", 'r') as stream: