Update app.py

app.py

@@ -3,6 +3,8 @@ import os
 import requests
 import io
 from PIL import Image
+from IPython.display import Audio, display
+from freeGPT import Client
 
 api_token = os.environ.get("API_TOKEN")
 API_URL = "https://api-inference.huggingface.co/models/facebook/musicgen-small"
@@ -14,6 +16,13 @@ st.write("Music generator using Facebook MusicGen, ChatGPT3 and Blip image capti
 img_prompt = st.file_uploader("Upload Image", type=["jpeg", "jpg", "png"])
 subm_btn = st.button("✨ Generate")
 
+def musquery(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.content
+
+def generate_audio(prompt):
+    return query({"inputs": prompt})
+
 if subm_btn and img_prompt is not None:
     def query(image_bytes):
         response = requests.post(API_URL_IMG, headers=headers, data=image_bytes)
@@ -25,5 +34,10 @@ if subm_btn and img_prompt is not None:
     image_bytes = image_bytes.getvalue()
 
     output = query(image_bytes)
-    prompt = output[0]['generated_text']
-    st.info(f"Generated Prompt for input image: {prompt}")
+    prompt_gpt = output[0]['generated_text']
+    prompt = Client.create_completion("gpt3", "Write a prompt a to generate music from this: " + prompt_gpt)
+    st.info(f"Generated Prompt for input image: {prompt}", icon="ℹ️")
+    music = generate_audio(prompt)
+    st.success('Music Generated Successfully!', icon="✅")
+    st.audio(music, format="audio/wav")
+    st.download_button("Download", music)