Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+st.set_page_config(page_title='Image Search App', layout='wide')
 import torch
 from transformers import AutoTokenizer, AutoModel
 import faiss
@@ -65,7 +66,6 @@ if st.button("Search"):
         # Display the images
         st.image(image_urls, width=200)
 
-if __name__ == '__main__':
-    st.set_page_config(page_title='Image Search App', layout='wide')
-    st.cache(allow_output_mutation=True)
-    run_app()
+# if __name__ == '__main__':    
+st.cache(allow_output_mutation=True)
+run_app()