基本格式
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,得到的页面效果图为: