Finishing touches

app.py

@@ -83,5 +83,5 @@ if st.session_state.emotion != None and st.session_state.genre != None and st.se
         st.text(st.session_state.text_prompt)
     if st.session_state.text_prompt:
        if st.button("Generate image from text description"):
-            image = image_service(st.session_state.text_prompt)
+            image = image_service(st.session_state.text_prompt).images[0]
             st.image(image)