Nikhil0987 commited on
Commit
c714dbd
·
verified ·
1 Parent(s): d58899f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -3,10 +3,8 @@ import torch
3
  from diffusers import ShapEPipeline
4
  from diffusers.utils import export_to_gif
5
 
6
-
7
- ckpt_id = "openai/shap-e"
8
- pipe = ShapEPipeline.from_pretrained(repo).to("cuda")
9
-
10
 
11
  guidance_scale = 15.0
12
  prompt = "a shark"
@@ -16,5 +14,4 @@ images = pipe(
16
  num_inference_steps=64,
17
  size=256,
18
  ).images
19
-
20
  gif_path = export_to_gif(images, "shark_3d.gif")
 
3
  from diffusers import ShapEPipeline
4
  from diffusers.utils import export_to_gif
5
 
6
+ ckpt_id = "openai/shap-e" # Model identifier from the Hugging Face Model Hub
7
+ pipe = ShapEPipeline.from_pretrained(ckpt_id).to("cuda") # Load the model correctly
 
 
8
 
9
  guidance_scale = 15.0
10
  prompt = "a shark"
 
14
  num_inference_steps=64,
15
  size=256,
16
  ).images
 
17
  gif_path = export_to_gif(images, "shark_3d.gif")