Update app.py

app.py

@@ -25,4 +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}")