Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 import torch
-from diffusers import DiffusionPipeline
+from diffusers import DiffusionPipeline, AutoPipelineForText2Image
 import base64
 from io import BytesIO
 
 def load_amused_model():
     # pipeline = DiffusionPipeline.from_pretrained("Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO") 
-    return DiffusionPipeline.from_pretrained("Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO") 
+    # AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo"
+    return AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo"
 
 # Generate image from prompt using AmusedPipeline
 def generate_image(prompt):