Update app.py

app.py

@@ -68,7 +68,7 @@ if uploaded_image is not None:
     st.image(image, caption="Uploaded Image", use_column_width=True)
     image_binary = image_bytes.getvalue()
     # Task prompt input
-    task_prompt = st.sidebar.text_input("Task Prompt", value="Describe the image in detail:")
+    task_prompt = st.sidebar.text_input("Task Prompt", value="<MORE_DETAILED_CAPTION>")
     # Additional text input (optional)
     text_input = st.sidebar.text_area("Input Questions", height=20)
     # Generate Caption button
@@ -77,9 +77,7 @@ if uploaded_image is not None:
         st.session_state.has_run = True
         st.write(task_prompt,"\n\n",text_input)
         inference.demo()
-        task_prompt = '<MORE_DETAILED_CAPTION>'
-        text_input="Describe this image in detail:" # kosmos-2 prompt implement here
         output=inference.run_example(image,st.session_state.model,st.session_state.processor,task_prompt,text_input)
-        st.write(output)
+        st.write(output.split(":")[1])