| import gradio as gr | |
| import diffusers | |
| def update(): | |
| try: | |
| pipeline.to("cuda") | |
| except: | |
| pass | |
| images= pipeline().images | |
| print(images) | |
| im = images[0] | |
| return im | |
| scheduler = diffusers.PNDMScheduler(num_train_timesteps=1100) | |
| pipeline = diffusers.PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler) | |
| css = """ | |
| .container { | |
| max-width: 512px; | |
| max-height: 512px; | |
| margin: auto; | |
| padding-top: 1.5rem; | |
| } | |
| """ | |
| block = gr.Blocks(css=css) | |
| with block as demo: | |
| with gr.Row(): | |
| out = gr.Image() | |
| btn = gr.Button("Generate") | |
| btn.click(update, inputs=None, outputs=out) | |
| demo.launch() | |