added setup.py

app.py

@@ -1,14 +1,22 @@
 import streamlit as st
 from main 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)
+    st.image(lorax, width = 250)
+
     st.header("\"" +search_request+ "\" reminds me of :")
     i = 0
     for row in range(0, 2):
         for col in st.columns(2) :
             image_name, comment = search_result[i]
-            col.image("img/"+image_name)
+            col.image("img/"+image_name, width = 300)
             col.header(comment)
             i = i + 1
     return

setup.py

@@ -0,0 +1,20 @@
+from huggingface_hub import hf_hub_url, cached_download
+from huggingface_hub.archive import unpack_archive
+
+
+# Specify the Google Drive link to the archive file
+archive_url = 'https://drive.google.com/uc?id=14QhofCbby053kWbVeWEBHCxOROQS-bjN'
+
+# Specify the destination directory within the Hugging Face space
+destination_dir = 'osanchik/flickr' 
+
+
+# Replace with your desired destination directory
+
+# Construct the destination path
+destination_path = hf_hub_url(destination_dir)
+# Download the archive to the destination path
+cached_download(archive_url, destination_path)
+# Unpack the archive
+unpack_archive(destination_path, destination_dir)
+print(f"Archive unpacked to: {destination_dir}")