File size: 642 Bytes
5edc0a2 b6cd8b0 5edc0a2 b6cd8b0 5edc0a2 d2980e7 5edc0a2 b6cd8b0 e81087c 77e2e9d 5edc0a2 4bf9eb4 e81087c 77e2e9d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import gradio as gr
from glyffuser_utils import GlyffuserPipeline
from diffusers import DPMSolverMultistepScheduler
pipeline = GlyffuserPipeline.from_pretrained("yuewu/glyffuser")
pipeline.scheduler = DPMSolverMultistepScheduler()
def infer(text, steps):
generated_images = pipeline(
text,
batch_size=1,
num_inference_steps=int(steps)
).images
return generated_images[0]
demo = gr.Interface(fn=infer, inputs=["text", "number"], outputs="image",
description="English text to Chinese glyph. Recommend <10 steps as this space is not running on GPU",
)
demo.launch() |