Update app.py

app.py

@@ -37,18 +37,20 @@ def main():
         with open(image_file, 'wb') as f:
             f.write(uploaded_file.getbuffer())
         predicted_class = predict_breed(image_file)
-        st.image(uploaded_file, caption=f'Predicted class: {predicted_class}', use_column_width=True)
+        st.image(uploaded_file, caption='', use_column_width=True)
+        st.write(f'Predicted class: **{predicted_class}**', unsafe_allow_html=True)
 
     # Add a button to trigger image upload
-    if st.button('Upload and Predict'):
+    if st.button('Upload and Predict '):
         uploaded_file = st.file_uploader('Choose an image of a dog', key='file_uploader_2', type=['jpg', 'jpeg', 'png'])
         if uploaded_file is not None:
             image_file = uploaded_file.name
             with open(image_file, 'wb') as f:
                 f.write(uploaded_file.getbuffer())
             predicted_class = predict_breed(image_file)
-            st.image(uploaded_file, caption=f'Predicted class: {predicted_class}', use_column_width=True)
-            
+            st.image(uploaded_file, caption='', use_column_width=True)
+            st.write(f'Predicted class: **{predicted_class}**', unsafe_allow_html=True)
+
 
 if __name__ == '__main__':
     main()