| import gradio as gr | |
| from PIL import Image | |
| from inference import single_image_wrapped | |
| # 调用 single_image_wrapped 函数的包装函数 | |
| def image_with_prompt_to_result(image, prompt): | |
| # 将图片传递给 single_image_wrapped 函数，得到输出结果 | |
| result = single_image_wrapped(image, prompt) | |
| return result | |
| # 创建 Gradio 接口 | |
| iface = gr.Interface( | |
| fn=image_with_prompt_to_result, # 调用的函数 | |
| inputs=[gr.Image(type="pil"), gr.Textbox(placeholder="输入提示词...")], # 输入：图片 + prompt | |
| outputs="text", # 输出：识别结果（文本） | |
| title="图片与提示词输入", # 界面标题 | |
| description="上传一张图片并输入提示词，获取识别结果", # 描述 | |
| ) | |
| # 启动应用 | |
| iface.launch() | |