Increased token limit

app.py

@@ -101,7 +101,8 @@ if uploaded_file is not None and uploaded_file != st.session_state.current_image
 
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
-        st.markdown(remove_code_blocks(message["content"]))
+        # st.markdown(remove_code_blocks(message["content"]))
+        st.markdown(message["content"])
 
 # === Chat Interface ===
 if prompt := st.chat_input("Ask a follow-up question..."):
@@ -125,7 +126,8 @@ if prompt := st.chat_input("Ask a follow-up question..."):
             else:
                 result = classifier.predict(image, user_input=prompt, reuse_embeddings=False)
 
-            st.markdown(remove_code_blocks(result["diagnosis"]))
+            # st.markdown(remove_code_blocks(result["diagnosis"]))
+            st.markdown(result["diagnosis"])
             st.session_state.messages.append({"role": "assistant", "content": result["diagnosis"]})
 
 if st.session_state.messages and st.button("📄 Download Chat as PDF"):