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(event):
-    jpgs = [f for f in os.listdir(IMAGE_FOLDER) if f.endswith(".jpg")]
+def get_random_image(event=None):
+    jpgs = [f for f in os.listdir(image_dir) if f.endswith(".jpg")]
     if not jpgs:
-        return STATIC_IMAGE_PATH, STATIC_IMAGE_PATH
+        return gr.update(value=STATIC_IMAGE_PATH), gr.update(value=STATIC_IMAGE_PATH)
+    
     selected = os.path.join(IMAGE_FOLDER, random.choice(jpgs))
-    return gr.update(value=selected)
+    print(selected)
+    return gr.update(value=selected), gr.update(value=selected)
 
         
 
@@ -50,7 +52,7 @@ with gr.Blocks() as demo:
 
     output = gr.Textbox(label="Model Output")
 
-    random_button.click(fn=get_random_image, outputs=image_display)
+    random_button.click(fn=get_random_image, outputs=[image_display, image_path])
     run_button.click(fn=process, inputs=[image_path, user_prompt], outputs=output)
 
 # with gr.Blocks() as demo: