Update setup.py

setup.py

@@ -2,19 +2,29 @@ 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'
+def downlad_images() :
 
-# Specify the destination directory within the Hugging Face space
-destination_dir = 'osanchik/flickr' 
+    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'
+    
+        # # 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}")
 
-
-# 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}")
+    return