Update app.py

app.py

@@ -11,7 +11,22 @@ from implement import *
 # from main import build_loaders
 # from CLIP import CLIPModel
 import os
+import zipfile
 
+# Define the filename
+zip_filename = 'Images.zip'
+
+# Check if the zip file exists in the current directory
+if os.path.isfile(zip_filename):
+    print(f"'{zip_filename}' found. Unzipping...")
+    
+    # Unzip the file
+    with zipfile.ZipFile(zip_filename, 'r') as zip_ref:
+        zip_ref.extractall()
+    print(f"'{zip_filename}' has been successfully unzipped.")
+else:
+    print(f"'{zip_filename}' not found in the current directory.")
+    
 with gr.Blocks(css="style.css") as demo:
     def get_image_embeddings(valid_df, model_path):
         tokenizer = DistilBertTokenizer.from_pretrained(CFG.text_tokenizer)