fffiloni commited on
Commit
901acde
1 Parent(s): 68b8f3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import torchaudio
3
  from audiocraft.models import MAGNeT
4
  from audiocraft.data.audio import audio_write
5
 
6
- model = MAGNeT.get_pretrained('facebook/magnet-small-10secs')
7
 
8
 
9
  def infer(description):
@@ -17,7 +17,7 @@ def infer(description):
17
  one_wav.cpu(),
18
  model.sample_rate,
19
  strategy="loudness",
20
- loudness_compressor=True)
21
 
22
  return "done"
23
 
 
3
  from audiocraft.models import MAGNeT
4
  from audiocraft.data.audio import audio_write
5
 
6
+ model = MAGNeT.get_pretrained('facebook/audio-magnet-medium')
7
 
8
 
9
  def infer(description):
 
17
  one_wav.cpu(),
18
  model.sample_rate,
19
  strategy="loudness",
20
+ loudness_compressor=False)
21
 
22
  return "done"
23