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() | |