Update app.py

app.py

@@ -6,6 +6,9 @@ from PIL import Image, ImageOps
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipeline = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")
 
+def get_select_index(evt: gr.SelectData):
+    return evt.index
+
 def squarify_image(img):
     if(img.height > img.width): bg_size = img.height
     else:  bg_size = img.width
@@ -76,13 +79,15 @@ with gr.Blocks() as demo:
         with gr.Column():
             version_gallery = gr.Gallery(label="Versions", type="pil", object_fit='contain')
             restore_button = gr.Button("Restore Version")
+            selected = gr.Number(show_label=False, visible=True)
             
         
     with gr.Row():
         out1 = gr.Image(format='png')
         out2 = gr.Image(format='png')
         out3 = gr.Image(format='png')
-        
+
+    version_gallery.select(get_select_index, None, selected)
     generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, version_gallery], outputs=[sketch_pad, out1, out2, out3, version_gallery])
 
 demo.launch()