DebasishDhal99 commited on
Commit
1d852fb
·
1 Parent(s): d67196e

fix: add spaces to comply with mandatory use og nvidia gpu on huggingface

Browse files
Files changed (2) hide show
  1. app.py +6 -0
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,4 +1,10 @@
1
  import gradio as gr
 
 
 
 
 
 
2
 
3
  with gr.Blocks(title="RoPE Explorer") as demo:
4
  with gr.Tabs():
 
1
  import gradio as gr
2
+ import spaces
3
+
4
+ @spaces.GPU
5
+ def gpu_test():
6
+ return "GPU available"
7
+
8
 
9
  with gr.Blocks(title="RoPE Explorer") as demo:
10
  with gr.Tabs():
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  gradio==6.26.0
2
 
3
  numpy>=2.0,<2.3
4
-
5
  plotly==7.0.0
 
1
  gradio==6.26.0
2
 
3
  numpy>=2.0,<2.3
4
+ spaces==0.51.3
5
  plotly==7.0.0