如何将Gradio图片上传按钮代码改为文件夹路径输入形式?
Gradio实现文件夹选择替代多图上传方案
你之前使用的gr.inputs.FilePicker属于Gradio旧版API(已被弃用),所以无法正常工作。以下是基于Gradio 3.x+版本的可行修改方案:
核心修改思路
用gr.Directory组件替代原有的UploadButton来获取文件夹路径,再在处理函数中筛选出文件夹内符合格式的图片文件。
完整示例代码
import gradio as gr import os def load_guide_frames(folder_path): # 筛选文件夹内指定格式的图片 if not folder_path: return [] valid_extensions = ('.png', '.jpg', '.jpeg') image_paths = [ os.path.join(folder_path, filename) for filename in os.listdir(folder_path) if filename.lower().endswith(valid_extensions) ] return image_paths with gr.Blocks() as demo: # 文件夹选择组件 reference_folder = gr.Directory(label="Select folder with Guide Frames") # 实时响应(对应原代码的live=True) reference_folder.change( fn=load_guide_frames, inputs=reference_folder, outputs=gr.Gallery(label="Loaded Guide Frames") ) demo.launch()
关键说明
gr.Directory是Gradio官方推荐的文件夹选择组件,返回选中的文件夹绝对路径- 处理函数中通过遍历文件夹+后缀筛选,实现原代码中
file_types的过滤效果 - 使用
reference_folder.change()事件实现实时响应,等同于原代码的live=True - 如果需要手动触发而非实时响应,可以替换为按钮点击事件(
gr.Button().click())
内容的提问来源于stack exchange,提问作者Furkan Gözükara
相关产品推荐
相关产品推荐

