Update app.py

app.py

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