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