fix n_cluster=1

app.py

@@ -4385,6 +4385,9 @@ with demo:
             images = [transform_image(image, resolution=(224, 224), stablediffusion=False) for image in resized_images]
             images = torch.stack(images)
             
+            if n_clusters == 1:
+                return np.zeros(len(images)), None, resized_images
+            
             is_cuda = torch.cuda.is_available()
             
             dino = torch.hub.load('facebookresearch/dino:main', 'dino_vitb16').eval()