modified import

app.py

@@ -1,14 +1,13 @@
 import streamlit as st
 from main import *
+from setup import *
+
 from PIL import Image
 
 def display(search_request, search_result) :
 
-
-
-    #put your own image here
-    lorax =  Image.open('img/Lorax.jpg')
-    print(lorax.width, lorax.height)
+    # lorax =  Image.open('img/Lorax.jpg')
+    # print(lorax.width, lorax.height)
     st.image(lorax, width = 250)
 
     st.header("\"" +search_request+ "\" reminds me of :")
@@ -22,6 +21,7 @@ def display(search_request, search_result) :
     return
 
 
+downlad_images()
 
 st.title('Find my pic!')
 

setup.py

@@ -1,3 +1,7 @@
+
+import os
+import streamlit as st
+
 from huggingface_hub import hf_hub_url, cached_download
 from huggingface_hub.archive import unpack_archive
 
@@ -7,8 +11,6 @@ def downlad_images() :
     img_dir = 'flickr'
     
     if not os.path.exists(img_dir) :
-        st.write("NEEED to download")
-        
         # os.makedirs(sample_dir, exist_ok=True)
         # # Specify the Google Drive link to the archive file
         # archive_url = 'https://drive.google.com/uc?id=14QhofCbby053kWbVeWEBHCxOROQS-bjN'