Update toxic1.py

toxic1.py

@@ -35,10 +35,10 @@ def toxicity_page():
 
         if proba >= 0.5:
             st.write(f'Степень токсичности комментария: {round(proba, 2)} – комментарий токсичный.')
-            st.image('maxresdefault.jpg', caption='Токсичный комментарий')
+            st.image('maxresdefault.jpg')
         else:
             st.write(f'Степень токсичности комментария: {round(proba, 2)} – комментарий не токсичный.')
-            st.image('c793397a-39df-5ff7-8137-e59568352c11.jpeg', caption='Токсичный комментарий')
+            st.image('c793397a-39df-5ff7-8137-e59568352c11.jpeg')
         st.write(f'Время предсказания: {prediction_time:.4f} секунд')
 
 toxicity_page()