gradio学习记录

基本格式

import gradio as gr
from wst_reasoning import predict_gradio

inputs = [
    gr.Textbox(placeholder="请输入",label="prompt"),
    gr.Dropdown(checkpoint, label="微调模型（可选）",placeholder="请选择，亦可不选"),
    gr.Slider(512, 1024,step=64,label="高度（可选）"),
    gr.Slider(512, 1024,step=64,label="宽度（可选）"),
    gr.Slider(1, 6,step=1,label="生成的图片数量（可选）")
]

demo = gr.Interface(
    fn=predict_gradio,
    inputs=inputs,
    outputs=["json"])

demo.launch(share=True)

gr.Interface()方法中fn为调用的函数，inputs为输入，outputs为输出。
输入输出与调用的函数的参数要保持一致，比如这个函数的参数为def predict_gradio(prompt=“一个金融管理系统的前端界面”, check_point=False, H=512, W=512, num_images_per_prompt=1)。
设置share=True可以在互联网上访问。

运行当前文件，就得到：

访问public URL，得到的页面效果图为：

输入输出