Create app.py

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+import torchaudio
+from audiocraft.models import MAGNeT
+from audiocraft. data. audio import audio_write
+
+model = MAGNeT.get_pretrained('facebook/magnet-small-10secs')
+
+
+def infer(description):
+    descriptions = ['disco beat', 'energetic EDM']
+
+    wav = model.generate(descriptions) 
+    
+    for idx, one_wav in enumerate(wav):
+        print(idx)
+        audio_write(f'{idx}', 
+                    one_wav.cpu(), 
+                    model.sample_rate, 
+                    strategy="loudness", 
+                    loudness_compressor=True)
+        
+    return "done"
+
+gr.Interface(
+    fn = infer,
+    inputs = gr.Textbox(value="gogo"),
+    outputs = gr.Textbox()
+).launch()