web demo share (text 2 img)
baiyutang opened this issue · 0 comments
baiyutang commented
import gradio as gr
from VisCPM import VisCPMPaint
# 修改你的模型地址
model_path = '/opt/ai/VisCPM/viscpm_paint_balance_checkpoint.pt'
painter = VisCPMPaint(model_path, image_safety_checker=False, prompt_safety_checker=False, add_ranker=True)
print("load image model success !")
def gen_img(txt, imgs):
image = painter.generate(txt)
imgs.append(image)
return "",imgs,imgs
with gr.Blocks() as demo:
imgs = gr.State([])
gallery = gr.Gallery(label="生成图片")
txt_message = gr.Textbox(label="输入文字")
txt_message.submit(gen_img, [txt_message, imgs], [txt_message, gallery,imgs])
demo.queue(concurrency_count=1, max_size=20).launch(share=False, debug=True, server_port=7866,
server_name="0.0.0.0")
以上内容来自群友分享,欢迎加入社群交流