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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -8,8 +8,10 @@ from io import BytesIO
8
 
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):
 
8
 
9
  def load_amused_model():
10
  # pipeline = DiffusionPipeline.from_pretrained("Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO")
11
+ # AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo")
12
+ # AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
13
+ return DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
14
+
15
 
16
  # Generate image from prompt using AmusedPipeline
17
  def generate_image(prompt):