Update app.py

app.py

@@ -52,7 +52,7 @@ def image_generator(description: str)->Image.Image :
     # output is a PIL.Image object
     image = client.text_to_image(
       prompt=description,
-      model="dreamlike-art/dreamlike-photoreal-2.0"
+      model="black-forest-labs/FLUX.1-schnell"
     )
     return image
   except Exception as e:
@@ -72,7 +72,7 @@ custom_role_conversions=None,
 
 
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)