Serhiy Stetskovych commited on
Commit
1f1a308
1 Parent(s): d48b9e1

Zero GPU support

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +2 -0
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 💻
4
  colorFrom: green
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 5.5.0
8
  app_file: app.py
9
  pinned: false
10
  ---
 
4
  colorFrom: green
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 4.44.1
8
  app_file: app.py
9
  pinned: false
10
  ---
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
  from audiosr import super_resolution, build_model
 
3
 
 
4
  def inference(audio_file, model_name, guidance_scale, ddim_steps):
5
  audiosr = build_model(model_name=model_name)
6
 
 
1
  import gradio as gr
2
  from audiosr import super_resolution, build_model
3
+ import spaces
4
 
5
+ @spaces.GPU
6
  def inference(audio_file, model_name, guidance_scale, ddim_steps):
7
  audiosr = build_model(model_name=model_name)
8