| import gradio as gr | |
| from diffusers import PNDMPipeline, PNDMScheduler | |
| scheduler = PNDMScheduler(num_train_timesteps=1100) | |
| pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler) | |
| def generate_img(): | |
| try: | |
| pipeline.to("cuda") | |
| except: | |
| pass | |
| images= pipeline().images | |
| im = images[0] | |
| return im | |
| iface = gr.Interface(fn=generate_img, inputs=[], outputs="image") | |
| iface.launch() |