Update app.py

app.py

@@ -6,24 +6,27 @@ from gradio_client import Client
 image_model = pipeline("image-classification", model="google/vit-base-patch16-224")
 
 def generate_music(prompt):
+    # Initialize the client with your API endpoint
     client = Client("https://haoheliu-audioldm-48k-text-to-hifiaudio-generation.hf.space/")
     
-    # Assuming the API requires these five arguments: prompt, duration, guidance_scale, seed, and num_waveforms
-    # Adjust the names and values according to the API's actual requirements
+    # Call the predict method with the correct parameters
     result = client.predict(
-        prompt=prompt,
-        duration=5,
-        guidance_scale=5.5,
-        seed=5,
-        num_waveforms=3,
-        # Remove the api_name if it's not expected/needed, or adjust accordingly
-        # api_name="/text2audio"  # Comment this out if api_name is not an expected argument
+        prompt,  # The main text input for your music generation
+        5,  # Duration in seconds
+        0,  # Guidance scale
+        5,  # Seed for generating music
+        1,  # Number of waveforms to generate
+        api_name="/text2audio"  # Specify the API name if required
     )
     
-    # Process the result
+    # Assuming the result includes the information you need directly
     print(result)
     return result
 
+# Example usage
+prompt = "A serene and peaceful melody to relax."
+music_result = generate_music(prompt)
+
 def generate_voice(prompt):
     # Tango API를 사용하여 음성 생성
     client = Client("https://declare-lab-tango.hf.space/")