cbensimon HF staff commited on
Commit
c4b86bc
·
verified ·
1 Parent(s): c4475c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -3,11 +3,8 @@ import open3d_zerogpu_fix
3
  import open3d as o3d
4
  import spaces
5
 
6
- assert o3d.core.cuda.is_available() is False
7
-
8
  @spaces.GPU
9
  def greet(n):
10
- assert o3d.core.cuda.is_available() is True
11
  return o3d.core.Tensor([], device=o3d.core.Device("CUDA:0")).device
12
 
13
  demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
 
3
  import open3d as o3d
4
  import spaces
5
 
 
 
6
  @spaces.GPU
7
  def greet(n):
 
8
  return o3d.core.Tensor([], device=o3d.core.Device("CUDA:0")).device
9
 
10
  demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())