Update app.py

app.py

@@ -33,7 +33,19 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-
+@tool
+def generate_music(prompt: str) -> str:
+    """
+    A tool that generates music based on the given prompt.
+    Args:
+        prompt: A string describing the style or mood of the music.
+    Returns:
+        A string representation or link to the generated music.
+    """
+    music_generator = pipeline("text-to-audio", model="facebook/musicgen-small")
+    music_output = music_generator(prompt, do_sample=True)
+    return str(music_output)
+    
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder: