fix image idx

app.py

@@ -3607,6 +3607,7 @@ with demo:
         def run_heatmap(images, eigvecs, image1_slider, prompt_image1, n_eig, distance_slider, flat_idx=None, overlay_image=True):
             gr.Info(f"current number of eigenvectors: {n_eig}")
             eigvecs = torch.tensor(eigvecs)
+            image1_slider = min(image1_slider, len(images)-1)
             images = [image[0] for image in images]
             if isinstance(images[0], str):
                 images = [Image.open(image[0]).convert("RGB").resize((256, 256)) for image in images]
@@ -3859,6 +3860,7 @@ with demo:
         def run_heatmap(images, eigvecs, image1_slider, prompt_image1, n_eig, distance_slider, flat_idx=None, overlay_image=True):
             gr.Info(f"current number of eigenvectors: {n_eig}", 2)
             eigvecs = torch.tensor(eigvecs)
+            image1_slider = min(image1_slider, len(images)-1)
             images = [image[0] for image in images]
             if isinstance(images[0], str):
                 images = [Image.open(image[0]).convert("RGB").resize((256, 256)) for image in images]