Update app.py

app.py

@@ -50,7 +50,7 @@ st.markdown(
     f'<img src="data:image/gif;base64,{data_url}" alt="cat gif" width=100%>',
     unsafe_allow_html=True,
 )
-st.image(image, caption='Training Steps')
+# st.image(image, caption='Training Steps')
 st.markdown("## Interactive Demo")
 
 # download the model: