Update app.py

app.py

@@ -2,17 +2,24 @@ from contextlib import nullcontext
 import gradio as gr
 import torch
 from torch import autocast
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionPipeline, StableDiffusionOnnxPipeline
 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 context = autocast if device == "cuda" else nullcontext
 dtype = torch.float16 if device == "cuda" else torch.float32
 
-pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/sd-naruto-diffusers", torch_dtype=dtype)
+if device == "cuda":
+    pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/sd-naruto-diffusers", torch_dtype=dtype)
+    
+else:
+    pipe = StableDiffusionOnnxPipeline.from_pretrained(
+        "lambdalabs/sd-naruto-diffusers",
+        revision="onnx",
+        provider="CPUExecutionProvider"
+    )
 pipe = pipe.to(device)
 
-
 # Sometimes the nsfw checker is confused by the Naruto images, you can disable
 # it at your own risk here
 disable_safety = True