AlexZaShared commited on
Commit
494f3dc
·
verified ·
1 Parent(s): 8e384af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -74,12 +74,14 @@ model = HfApiModel(
74
  # temperature=0.6,
75
  #)
76
 
 
 
77
  # Import tool from Hub
78
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
79
 
80
  with open("prompts.yaml", 'r') as stream:
81
  prompt_templates = yaml.safe_load(stream)
82
-
83
  agent = CodeAgent(
84
  model=model,
85
  tools=[final_answer], ## add your tools here (don't remove final answer)
@@ -92,5 +94,6 @@ agent = CodeAgent(
92
  prompt_templates=prompt_templates
93
  )
94
 
 
95
 
96
  GradioUI(agent).launch()
 
74
  # temperature=0.6,
75
  #)
76
 
77
+ print("Model Image")
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)
84
+
85
  agent = CodeAgent(
86
  model=model,
87
  tools=[final_answer], ## add your tools here (don't remove final answer)
 
94
  prompt_templates=prompt_templates
95
  )
96
 
97
+ print("Start GradioUI")
98
 
99
  GradioUI(agent).launch()