response jsonify

app.py

@@ -76,8 +76,9 @@ if st.button("Generate Audio") and genre and energy_level and description and te
         payload = {"inputs": {"prompt": prompt, "duration": duration, "track": audio_base64}}
     st.text("Generating audio...")
     response = requests.post(API_URL, headers=headers, json=payload)
-    st_state.audio = np.array(response.json()[0]['generated_audio'], dtype=np.float32)
-    sample_rate = response.json()[0]['sample_rate']
+    response = response.json()[0]
+    st_state.audio = np.array(response['generated_audio'], dtype=np.float32)
+    sample_rate = response['sample_rate']
     st.audio(st_state.audio, format="audio/wav", sample_rate=sample_rate, start_time=0)
 
 # Post-processing options