jkorstad commited on
Commit
c29fe64
Β·
verified Β·
1 Parent(s): e4a26af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
  import numpy as np
3
  import torch
 
 
4
  from diffusers import DiffusionPipeline
5
  from PIL import Image
6
 
@@ -11,7 +13,7 @@ multi_view_diffusion_pipeline = DiffusionPipeline.from_pretrained(
11
  trust_remote_code=True,
12
  ).to("cuda")
13
 
14
-
15
  def run(image):
16
  image = np.array(image, dtype=np.float32) / 255.0
17
  images = multi_view_diffusion_pipeline(
 
1
  import gradio as gr
2
  import numpy as np
3
  import torch
4
+ import spaces
5
+
6
  from diffusers import DiffusionPipeline
7
  from PIL import Image
8
 
 
13
  trust_remote_code=True,
14
  ).to("cuda")
15
 
16
+ @spaces.GPU
17
  def run(image):
18
  image = np.array(image, dtype=np.float32) / 255.0
19
  images = multi_view_diffusion_pipeline(