Update app.py

app.py

@@ -17,7 +17,11 @@ def create_new_vday_card_tool(theme:str) -> str: #it's import to specify the ret
     """
     try:
         # Import tool from Hub
-        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+        image_generation_tool = Tool.from_space(
+            space_id="black-forest-labs/FLUX.1-schnell",
+            name="generate-image",
+            description="Generate image from prompt"
+        )
         prompt = "A high-resolution image of a valentine's day card themed around " + theme
         image = image_generation_tool(prompt)
         return final_answer(image)