Geek7 commited on
Commit
6d1caec
1 Parent(s): 4ae6013

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import dask
4
  from dask import delayed
5
 
6
  # Load model
7
- pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4")
8
 
9
  def generate_image(prompt, num_inference_steps=50):
10
  """
 
4
  from dask import delayed
5
 
6
  # Load model
7
+ pipe = DiffusionPipeline.from_pretrained("kandinsky-community/kandinsky-2-1")
8
 
9
  def generate_image(prompt, num_inference_steps=50):
10
  """