Nymbo commited on
Commit
1e9aa1e
1 Parent(s): 9b38c9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,14 +2,14 @@ import torch
2
  import gradio as gr
3
  import torchaudio
4
  from transformers import AutoModel
5
- import spaces
6
 
7
 
8
  checkpoint_path = "./"
9
  model = AutoModel.from_pretrained(checkpoint_path, trust_remote_code=True)
10
 
11
 
12
- @spaces.GPU()
13
  def restore_audio(input_audio):
14
  # Load the audio file
15
  waveform, sample_rate = torchaudio.load(input_audio)
 
2
  import gradio as gr
3
  import torchaudio
4
  from transformers import AutoModel
5
+ # import spaces
6
 
7
 
8
  checkpoint_path = "./"
9
  model = AutoModel.from_pretrained(checkpoint_path, trust_remote_code=True)
10
 
11
 
12
+ # @spaces.GPU()
13
  def restore_audio(input_audio):
14
  # Load the audio file
15
  waveform, sample_rate = torchaudio.load(input_audio)