Update app.py

app.py

@@ -1,53 +1,3 @@
-import os
-import zipfile
-
-import os
-import zipfile
-
-# Define the filename
-zip_filename = 'Images.zip'
-
-# Get the current directory path
-current_directory = os.getcwd()
-print(f"Current directory: {current_directory}")
-
-# Append a custom string to the current directory path (for demonstration)
-custom_directory = os.path.join(current_directory, 'UnzippedContent')
-
-# Ensure the custom directory exists
-os.makedirs(custom_directory, exist_ok=True)
-
-# Print the contents of the current directory before unzipping
-print(f"Contents of current directory before unzipping: {os.listdir(current_directory)}")
-
-# Check if the zip file exists in the current directory
-zip_file_path = os.path.join(current_directory, zip_filename)
-if os.path.isfile(zip_file_path):
-    # Open the zip file
-    with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
-        # Extract all contents of the zip file to the custom directory
-        zip_ref.extractall(custom_directory)
-    print(f"'{zip_filename}' has been successfully unzipped to '{custom_directory}'.")
-
-    # Print the contents of the custom directory after unzipping
-    print(f"Contents of '{custom_directory}': {os.listdir(custom_directory)}")
-else:
-    print(f"'{zip_filename}' not found in the current directory.")
-
-# Print the contents of the current directory after unzipping
-print(f"Contents of current directory after unzipping: {os.listdir(current_directory)}")
-
-# Code to list the contents in the application (e.g., web interface)
-def list_unzipped_contents():
-    unzipped_contents = os.listdir(custom_directory)
-    print(f"Unzipped contents: {unzipped_contents}")
-    return unzipped_contents
-
-# Call the function to list unzipped contents (this part should be integrated with your app's display logic)
-unzipped_files = list_unzipped_contents()
-# Integrate `unzipped_files` with your app's interface to display the contents
-
-
 import gradio as gr
 import gc
 import cv2
@@ -57,15 +7,12 @@ from tqdm import tqdm
 from transformers import DistilBertTokenizer
 import matplotlib.pyplot as plt
 from implement import *
-# import config as CFG
-# from main import build_loaders
-# from CLIP import CLIPModel
+import config as CFG
+from main import build_loaders
+from CLIP import CLIPModel
 import os
-
-
-
-
-    
+os.environ['HTTPS_PROXY']="http://185.46.212.90:80/"
+os.environ['HTTP_PROXY']="http://185.46.212.90:80/"
 with gr.Blocks(css="style.css") as demo:
     def get_image_embeddings(valid_df, model_path):
         tokenizer = DistilBertTokenizer.from_pretrained(CFG.text_tokenizer)