Update app.py

app.py

@@ -24,29 +24,18 @@ def imgquery(payload):
     return response.content
 
 def generate_audio(prompt):
-    audio_bytes = query({"inputs": prompt})
-    
-    bytes_io = io.BytesIO()
-    bytes_io.write(audio_bytes)
-    bytes_io.seek(0)
-    
-    display(Audio(data=bytes_io.read(), autoplay=False))
-    
-    return bytes_io
+    return query({"inputs": prompt})
 
 def generate_image(prompt):
-    image_bytes = imgquery({
-        "inputs": prompt,
-    })
-    image = Image.open(io.BytesIO(image_bytes))
-    return image
+    image_bytes = imgquery({"inputs": prompt})
+    return Image.open(io.BytesIO(image_bytes))
 
 if generate_btn:
     audio1 = generate_audio(prompt)
     audio2 = generate_audio(prompt)
     audio3 = generate_audio(prompt)
 
-    title1 = Client.create_completion("gpt3", "generate name of song that have this description. In answer give onli short title" + prompt)
+    title1 = Client.create_completion("gpt3", "generate name of song that have this description. In answer give only short title" + prompt)
     title2 = Client.create_completion("gpt3", "generate name of song that have this description. In answer give only short title" + prompt)
     title3 = Client.create_completion("gpt3", "generate name of song that have this description. In answer give just short title" + prompt)
 
@@ -54,17 +43,14 @@ if generate_btn:
     cover2 = generate_image("Generate a cover for this song:" + prompt)
     cover3 = generate_image("Generate a thumbnail for this song:" + prompt)
 
-    audio1, audio1, audio3 = st.columns(3)
+    st.image(cover1)
+    st.header(title1)
+    st.audio(audio1, format="audio/wav")
 
-    with audio1:
-        st.image(cover1)
-        st.header(title1)
-        st.audio(audio1, format="audio/wav")
-    with audio2:
-        st.image(cover2)
-        st.header(title2)
-        st.audio(audio2, format="audio/wav")
-    with audio3:
-        st.image(cover3)
-        st.header(title3)
-        st.audio(audio3, format="audio/wav")
+    st.image(cover2)
+    st.header(title2)
+    st.audio(audio2, format="audio/wav")
+
+    st.image(cover3)
+    st.header(title3)
+    st.audio(audio3, format="audio/wav")