musicgen-large / call_endpoint.py
onurio's picture
working handler
e657366
raw
history blame contribute delete
722 Bytes
import requests
# Define the URL of the FastAPI server
url = "http://localhost:8000/generate_audio"
# Define the text for which you want to generate audio
text = "lo-fi music with a soothing melody"
# Define the headers for the request
headers = {"Content-Type": "application/json"}
# Make a POST request to the endpoint with the text data in the request body and the specified header
response = requests.post(url, json={"text": text}, headers=headers)
# Check if the request was successful
if response.status_code == 200:
# Save the audio file
with open("generated_audio.wav", "wb") as f:
f.write(response.content)
print("Audio file saved successfully.")
else:
print("Error:", response.text)