File size: 578 Bytes
ef12b6d 2a9d621 ef12b6d 23cfb8e 99a019a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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) |