| import gradio as gr | |
| def greet(name): | |
| return "Hello " + name + "!!" | |
| from diffusers import AutoPipelineForText2Image | |
| import torch | |
| def pipee(name): | |
| pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16).to("cpu") | |
| pipeline.load_lora_weights("ostris/super-cereal-sdxl-lora", weight_name="cereal_box_sdxl_v1.safetensors") | |
| prompt = "bears, pizza bites" | |
| image = pipeline(prompt).images[0] | |
| return image | |
| iface = gr.Interface(fn=pipee, inputs="text", outputs="text") | |
| iface.queue().launch(debug=True) |