Update app.py

app.py

@@ -26,7 +26,8 @@ def run_vlm(image, prompt):
 
 def random_image():
     path = random.choice(images)
-    return path # Also return path if needed for model input
+    return Image.open(path), path
+
 
 def process(image_path, user_prompt):
     image = Image.open(image_path)
@@ -37,11 +38,13 @@ with gr.Blocks() as demo:
         image_display = gr.Image(type="filepath", label="Selected Image")
         random_button = gr.Button("Randomize Photo")
 
+    image_path = gr.Textbox(visible=False)
     user_prompt = gr.Textbox(label="User Prompt")
     run_button = gr.Button("Run Model")
     output = gr.Textbox(label="Model Output")
 
-    random_button.click(fn=random_image, outputs=image_display)
-    run_button.click(fn=process, inputs=[image_display, user_prompt], outputs=output)
+    random_button.click(fn=random_image, outputs=[image_display, image_path])
+    run_button.click(fn=process, inputs=[image_path, user_prompt], outputs=output)
+
 
 demo.launch()