Create Main.py

Main.py

@@ -0,0 +1,19 @@
+import torch
+from diffusers import ShapEPipeline
+from diffusers.utils import export_to_gif
+
+
+ckpt_id = "openai/shap-e"
+pipe = ShapEPipeline.from_pretrained(repo).to("cuda")
+
+
+guidance_scale = 15.0
+prompt = "a shark"
+images = pipe(
+    prompt,
+    guidance_scale=guidance_scale,
+    num_inference_steps=64,
+    size=256,
+).images
+
+gif_path = export_to_gif(images, "shark_3d.gif")