Update app.py

app.py

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