Model loading once

app.py

@@ -4,11 +4,13 @@ from audiocraft.data.audio import audio_write
 import os
 import gradio as gr
 
+model = AudioGen.get_pretrained('facebook/audiogen-medium')
+model.set_generation_params(duration=5)  # generate 5 seconds.
+
 def generate_audio(descriptions):
     if not os.path.exists('audio_files'):
         os.makedirs('audio_files')
-    model = AudioGen.get_pretrained('facebook/audiogen-medium')
-    model.set_generation_params(duration=5)  # generate 5 seconds.
+    
     wav = model.generate([descriptions])  # generates 3 samples.
     results = []