RanM commited on
Commit
29df8e1
·
verified ·
1 Parent(s): 8d9c41d

Update app.py

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