Update app.py

app.py

@@ -4,7 +4,7 @@ from diffusers.utils import export_to_gif
 
 
 ckpt_id = "openai/shap-e"
-pipe = ShapEPipeline.from_pretrained(repo).to("cuda")
+pipe = ShapEPipeline.from_pretrained(repo).to("cpu")
 
 
 guidance_scale = 15.0
@@ -12,7 +12,7 @@ prompt = "a shark"
 images = pipe(
     prompt,
     guidance_scale=guidance_scale,
-    num_inference_steps=64,
+    num_inference_steps=16,
     size=256,
 ).images