Update app.py

app.py

@@ -36,6 +36,7 @@ if subm_btn and img_prompt is not None:
     output = query(image_bytes)
     prompt_gpt = output[0]['generated_text']
     prompt = Client.create_completion("gpt3", "Write a short, very short prompt a to generate music from this: " + prompt_gpt)
+    st.image(image_bytes, caption=prompt)
     st.caption(f'Generated prompt for input image - :blue[{prompt}]')
     music = generate_audio(prompt)
     st.success('Music Generated Successfully!', icon="✅")