Update app.py

app.py

@@ -1,11 +1,11 @@
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionXLPipeline
 import torch
 from gradio import Interface, Image
 import gradio as gr
 import spaces
 
 model_id = "RunDiffusion/Juggernaut-X-v10"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 
 @spaces.GPU()