Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from diffusers import StableDiffusion3Pipeline
+from diffusers import StableDiffusion
 import gradio as gr
 import spaces
 
@@ -11,7 +11,7 @@ else:
     print("Using CPU")
 
 # Initialize the pipeline and download the model
-pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium", torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium", torch_dtype=torch.float16, use_auth_token=HF_TOKEN)
 pipe.to(device)
 
 # Define the image generation function