Test text color

app.py

@@ -14,7 +14,7 @@ st.markdown('We are using a sample fo the [TinyStories](roneneldan/TinyStories)
 text_text = dataset['train'][10]['text']
 st.write(text_text)
 st.markdown(':red[' + text_text + ']' )
-st.markdown("The next word is <span style='color:red'>red</span>",unsafe_allow_html=True)
+st.markdown("The next word is <span style='color:red'>" + text_text + "</span>",unsafe_allow_html=True)
 
 st.markdown('The threshold changes the level of connectivity in the network. The reange is from 0 (less similar) to 1 (completely similar)')
 threshhold = st.slider('Threshhold',0.0,1.0,step=0.1)