Update app.py

app.py

@@ -4,15 +4,21 @@ from diffusers.utils import export_to_gif
 
 ckpt_id = "openai/shap-e"
 device = torch.device("cpu")
-pipe = ShapEPipeline.from_pretrained(ckpt_id, use_auth_token=True).to(device)
+
+# Attempt to load the pipeline without auth for testing
+try:
+    pipe = ShapEPipeline.from_pretrained(ckpt_id).to(device)
+except Exception as e:
+    print(f"Error loading model: {e}")
+    exit(1)
 
 guidance_scale = 15.0
-prompt = "A gentle Ai voice assistant constructed from a circle ring and 3 lines that fly alongside the circle"
-images = pipe(
-    prompt,
-    guidance_scale=guidance_scale,
-    num_inference_steps=16,
-    size=256,
-).images
+prompt = "A gentle AI voice assistant constructed from a circle ring and 3 lines that fly alongside the circle"
 
-gif_path = export_to_gif(images, "assistant_3d.gif")
+# Run the model with minimal parameters to troubleshoot
+try:
+    images = pipe(prompt, guidance_scale=guidance_scale).images
+    gif_path = export_to_gif(images, "assistant_3d.gif")
+    print(f"GIF created at: {gif_path}")
+except Exception as e:
+    print(f"Error generating images: {e}")