Kunhak Lee commited on
Commit
820b693
1 Parent(s): f9f119e
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  import torch
 
3
  from diffusers import DiffusionPipeline
4
  import numpy as np
5
 
@@ -8,7 +9,7 @@ multi_view_diffusion_pipeline = DiffusionPipeline.from_pretrained(
8
  custom_pipeline="dylanebert/multi-view-diffusion",
9
  torch_dtype=torch.float16,
10
  trust_remote_code=True,
11
- ).to("cuda")
12
 
13
  def run(image):
14
  image = np.array(image, dtype=np.float32) / 255.0
 
1
  import gradio as gr
2
  import torch
3
+ from PIL import Image
4
  from diffusers import DiffusionPipeline
5
  import numpy as np
6
 
 
9
  custom_pipeline="dylanebert/multi-view-diffusion",
10
  torch_dtype=torch.float16,
11
  trust_remote_code=True,
12
+ ).to("cpu")
13
 
14
  def run(image):
15
  image = np.array(image, dtype=np.float32) / 255.0