Update app.py

app.py

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