improve

app.py

@@ -36,10 +36,11 @@ with tabs2:
     file_name = st.file_uploader("Upload an image to caption",key="caption_image")
 
     if file_name is not None:
+        col1, col2 = st.columns(2)
         image = Image.open(file_name)
-        st.image(image, use_column_width=True)
+        col1.image(image, use_column_width=True)
         captioner = get_model_image_captioning()
         predictions = captioner(image)
-        st.write("**generated_text: **")
+        col2.header("generated caption")
         for generated_text in predictions:
             st.write(f"\n{generated_text['generated_text']}")