Masrkai commited on
Commit
e26f3fa
1 Parent(s): 0962f46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -1,16 +1,13 @@
1
  import torch
2
- from transformers import ShapEPipeline
3
-
4
  from diffusers import ShapEPipeline
5
  from diffusers.utils import export_to_gif
6
 
7
- ckpt_id = "openai/shap-e-xs"
8
  device = torch.device("cpu")
9
-
10
  pipe = ShapEPipeline.from_pretrained(ckpt_id, use_auth_token=True).to(device)
11
 
12
  guidance_scale = 15.0
13
- prompt = "a shark"
14
  images = pipe(
15
  prompt,
16
  guidance_scale=guidance_scale,
@@ -18,4 +15,4 @@ images = pipe(
18
  size=256,
19
  ).images
20
 
21
- gif_path = export_to_gif(images, "shark_3d.gif")
 
1
  import torch
 
 
2
  from diffusers import ShapEPipeline
3
  from diffusers.utils import export_to_gif
4
 
5
+ ckpt_id = "openai/shap-e"
6
  device = torch.device("cpu")
 
7
  pipe = ShapEPipeline.from_pretrained(ckpt_id, use_auth_token=True).to(device)
8
 
9
  guidance_scale = 15.0
10
+ prompt = "A gentle Ai voice assistant constructed from a circle ring and 3 lines that fly alongside the circle"
11
  images = pipe(
12
  prompt,
13
  guidance_scale=guidance_scale,
 
15
  size=256,
16
  ).images
17
 
18
+ gif_path = export_to_gif(images, "assistant_3d.gif")