Leeroy commited on
Commit
1a56e68
·
1 Parent(s): 15ca958

Fix ZeroGPU runtime check by adding @spaces.GPU probe function

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -1,5 +1,11 @@
1
  import gradio as gr
2
  import random
 
 
 
 
 
 
3
 
4
  _TITLE = '''
5
  # Sparc3D: Next-Gen High-Resolution 3D Model Generation
 
1
  import gradio as gr
2
  import random
3
+ import spaces
4
+
5
+
6
+ @spaces.GPU
7
+ def _hf_zerogpu_probe():
8
+ return None
9
 
10
  _TITLE = '''
11
  # Sparc3D: Next-Gen High-Resolution 3D Model Generation