Update app.py

app.py

@@ -53,17 +53,17 @@ if st.session_state.image_details:
     # Chat input
     prompt = st.chat_input("Ask something about the image")
     if prompt:
-        # Add user message to session state
         st.session_state.messages.append({"role": "user", "content": prompt})
-
-        # Generate response using Qwen
+    
+        # Generate response
         response = generate_response([st.session_state.image_details], prompt)
         st.session_state.messages.append({"role": "assistant", "content": response})
-
+    
         # Display user message
         with st.chat_message("user"):
             st.write(prompt)
-
+    
         # Display assistant message
         with st.chat_message("assistant"):
             st.write(response)
+