ALeLacheur commited on
Commit
f98b4e7
·
verified ·
1 Parent(s): 7b6ebbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def predict(inp):
45
  encoders = [EncodecModel.from_pretrained("facebook/encodec_48khz")]
46
 
47
  #Run modelD to disguise audio
48
- with torch.no_grad():
49
  waveformD = attack.poison_audio(waveform, sample_rate, encoders)
50
 
51
  #Transform output audio into gradio-readable format
 
45
  encoders = [EncodecModel.from_pretrained("facebook/encodec_48khz")]
46
 
47
  #Run modelD to disguise audio
48
+ with torch():
49
  waveformD = attack.poison_audio(waveform, sample_rate, encoders)
50
 
51
  #Transform output audio into gradio-readable format