Spaces:

AyushDey
/

Image_Generator

Runtime error

AyushDey commited on Jul 28, 2023

Commit

019c67b

•

1 Parent(s): c1ba13f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 from diffusers import DiffusionPipeline
-pipe = DiffusionPipeline.from_pretrained('runwayml/stable-diffusion-v1-5')
 def imagen(prompt):
     return pipe(prompt).images[0]

 from transformers import pipeline
 from diffusers import DiffusionPipeline
+pipe = DiffusionPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0')
 def imagen(prompt):
     return pipe(prompt).images[0]