Update app.py

app.py

@@ -25,5 +25,5 @@ if subm_btn and img_prompt is not None:
     image_bytes = image_bytes.getvalue()
 
     output = query(image_bytes)
-    #prompt = output['generated_text']
-    st.info(f"Generated Prompt for input image: {output}")
+    prompt = output[0]['generated_text']
+    st.info(f"Generated Prompt for input image: {prompt}")