Update app.py

app.py

@@ -54,6 +54,8 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7, use_bfloat=
         model.to(torch.bfloat16)
     if use_eval==True:
         model.eval()
+    torch.cuda.empty_cache()
+    torch.cuda.reset_peak_memory_stats()
     output = generate_diffusion_cond(
         model,
         steps=steps,