RoyYang0714 commited on
Commit
e6cd066
·
1 Parent(s): 64a70d8

fix: Add back spaces.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  """Gradio Demo for 3D-MOOD."""
2
 
 
3
  import gradio as gr
4
  import gc
5
  import os
@@ -79,7 +80,7 @@ def get_3d_mood_swin_base(
79
  depth_head=depth_head,
80
  )
81
 
82
-
83
  def run_3d_mood(fx, fy, cx, cy, image):
84
  """Run 3D-MOOD demo."""
85
 
 
1
  """Gradio Demo for 3D-MOOD."""
2
 
3
+ import spaces
4
  import gradio as gr
5
  import gc
6
  import os
 
80
  depth_head=depth_head,
81
  )
82
 
83
+ @spaces.GPU
84
  def run_3d_mood(fx, fy, cx, cy, image):
85
  """Run 3D-MOOD demo."""
86