Update app.py

app.py

@@ -16,9 +16,6 @@ split_dataset = dataset['train'].train_test_split(test_size=0.2, seed=42)  # 80%
 dataset['train'] = split_dataset['train']
 dataset['test'] = split_dataset['test']
 
-num_clusters = 10  # Choose the number of clusters
-kmeans = KMeans(n_clusters=num_clusters, random_state=42)
-kmeans.fit(encoded_images)
 
 def create_url_from_title(title: str, timestamp: int):
     video_urls = load_dataset("eybro/video_urls")
@@ -68,10 +65,8 @@ def process_image(image):
     return pooled_array  # Shape: (1, n_features)
     
 def inference(image):
-    """"""
     input_image = process_image(image)
 
-    # input_image = encoded_images[2000]
     
     nearest_neighbors = find_nearest_neighbors(encoded_images, input_image, top_n=5)