Update app.py

app.py

@@ -16,19 +16,26 @@ 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.")
+import os
+import zipfile
+
+
+
     
 with gr.Blocks(css="style.css") as demo:
     def get_image_embeddings(valid_df, model_path):
+        # Define the filename
+        zip_filename = 'Images.zip'
+        
+        # Check if the file exists
+        if os.path.isfile(zip_filename):
+            # Open the zip file
+            with zipfile.ZipFile(zip_filename, 'r') as zip_ref:
+                # Extract all contents of the zip file to the current directory
+                zip_ref.extractall()
+            print(f"'{zip_filename}' has been successfully unzipped.")
+        else:
+            print(f"'{zip_filename}' not found in the current directory.")
         tokenizer = DistilBertTokenizer.from_pretrained(CFG.text_tokenizer)
         valid_loader = build_loaders(valid_df, tokenizer, mode="valid")