Replace columns by beta_columns

app.py

@@ -67,7 +67,7 @@ def main():
               </style>''',
               unsafe_allow_html=True)
   st.sidebar.markdown(description)
-  _, col1, col2, _ = st.columns([2, 10, 2, 2])
+  _, col1, col2, _ = st.beta_columns([2, 10, 2, 2])
   query = col1.text_input('')
   corpus = col2.radio('', ["Unsplash","Movies"])
   if len(query) > 0: