Update app.py

app.py

@@ -24,12 +24,14 @@ def run_vlm(image, prompt):
     return processor.decode(out[0], skip_special_tokens=True)
 # return f"Model output for: '{prompt}' and selected image."
 
-def get_random_image():
+def get_random_image(event):
     jpgs = [f for f in os.listdir(IMAGE_FOLDER) if f.endswith(".jpg")]
     if not jpgs:
         return STATIC_IMAGE_PATH, STATIC_IMAGE_PATH
     selected = os.path.join(IMAGE_FOLDER, random.choice(jpgs))
-    return selected, selected 
+    return gr.update(value=selected)
+
+        
 
 def process(image_path, user_prompt):
     image = Image.open(image_path)
@@ -48,7 +50,7 @@ with gr.Blocks() as demo:
 
     output = gr.Textbox(label="Model Output")
 
-    random_button.click(fn=get_random_image, outputs=[image_display, image_path])
+    random_button.click(fn=get_random_image, outputs=image_display)
     run_button.click(fn=process, inputs=[image_path, user_prompt], outputs=output)
 
 # with gr.Blocks() as demo: