Spaces:
Runtime error
Runtime error
import gradio as gr | |
import cv2 | |
from modelscope.pipelines import pipeline | |
from modelscope.utils.constant import Tasks | |
text2image = pipeline(Tasks.text_to_image_synthesis,"damo/cv_diffusion_text-to-image-synthesis_tiny") | |
def inference(text): | |
result = text2image({"text":text}) | |
cv2.imwrite("result.png", result["output_imgs"][0]) | |
return "result.png" | |
title = "chinese stable diffusion" | |
gr.Interface(inference, "text","image", title=title).launch() | |