Update app.py

app.py

@@ -6,19 +6,22 @@ from gradio_client import Client
 image_model = pipeline("image-classification", model="google/vit-base-patch16-224")
 
 def generate_music(prompt):
-    # audioldm API 사용하여 음악 생성 API 호출
     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
     result = client.predict(
-        "playing piano.",	# str in 'Input text' Textbox component
-        "Low quality.",	# str in 'Negative prompt' Textbox component
-        5,	# int | float (numeric value between 5 and 15) in 'Duration (seconds)' Slider component
-        5.5,	# int | float (numeric value between 0 and 7) in 'Guidance scale' Slider component
-        5,	# int | float in 'Seed' Number component
-        3,	# int | float (numeric value between 1 and 5) in 'Number waveforms to generate' Slider component
-        api_name="/text2audio"
+        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
     )
+    
+    # Process the result
     print(result)
-    #audio_result = extract_audio(result)
     return result
 
 def generate_voice(prompt):