make sure there is a prompt

app.py

@@ -85,6 +85,9 @@ assert len(images) == 34, "There should be exactly 34 images in the 'images' fol
 async def select_fn(data: gr.SelectData, prompt: str):
     global selected_image_index
     selected_image_index = data.index
+    if prompt == "":
+        print("Prompt is empty, returning original image")
+        return images[selected_image_index]
     return await predict(prompt) 
     
 async def predict(prompt):