RanM commited on
Commit
cabaf0b
·
verified ·
1 Parent(s): 9e09422

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import base64
5
  from io import BytesIO
6
 
7
  def load_amused_model():
8
- return DiffusionPipeline.from_pretrained("amused/amused-256")
9
 
10
  # Generate image from prompt using AmusedPipeline
11
  def generate_image(prompt):
 
5
  from io import BytesIO
6
 
7
  def load_amused_model():
8
+ return DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers")
9
 
10
  # Generate image from prompt using AmusedPipeline
11
  def generate_image(prompt):