RanM commited on
Commit
6ad6199
·
verified ·
1 Parent(s): 6d1d03a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from io import BytesIO
9
  def load_amused_model():
10
  # pipeline = DiffusionPipeline.from_pretrained("Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO")
11
  # AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo"
12
- return AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo")
13
 
14
  # Generate image from prompt using AmusedPipeline
15
  def generate_image(prompt):
 
9
  def load_amused_model():
10
  # pipeline = DiffusionPipeline.from_pretrained("Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO")
11
  # AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo"
12
+ return AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
13
 
14
  # Generate image from prompt using AmusedPipeline
15
  def generate_image(prompt):