fix bugs

app.py

@@ -38,10 +38,14 @@ with tabs2:
     if file_name is not None:
         col1, col2 = st.columns(2)
         image = Image.open(file_name)
+
         col1.image(image, use_column_width=True)
+
         captioner = get_model_image_captioning()
-        col2.header("generated caption")
-        with col2.spinner("Generating caption..."):
-            predictions = captioner(image)
-        for generated_text in predictions:
-            col2.write(f"\n{generated_text['generated_text']}")
+
+        with col2:
+            st.header("generated caption")
+            with st.spinner("Generating caption..."):
+                predictions = captioner(image)
+            for generated_text in predictions:
+                st.write(f"\n{generated_text['generated_text']}")