Update app.py

app.py

@@ -36,7 +36,7 @@ print(f"Using device: {device}")
 model.to(device,torch.float32)
 
 @spaces.GPU(duration=60)
-def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7, use_bfloat=False):
+def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7, use_bfloat=False, use_eval=False):
     print(f"Prompt received: {prompt}")
     print(f"Settings: Duration={seconds_total}s, Steps={steps}, CFG Scale={cfg_scale}")
     sample_rate = model_config["sample_rate"]
@@ -52,6 +52,8 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7, use_bfloat=
     print("Generating audio...")
     if use_bfloat==True:
         model.to(torch.bfloat16)
+    if use_eval==True:
+        model.eval()
     output = generate_diffusion_cond(
         model,
         steps=steps,
@@ -88,7 +90,8 @@ interface = gr.Interface(
         gr.Slider(0, 420, value=30, label="Duration in Seconds"),
         gr.Slider(10, 420, value=100, step=10, label="Number of Diffusion Steps"),
         gr.Slider(1.0, 32.0, value=7.0, step=0.1, label="CFG Scale"),
-        gr.Checkbox(value=False, label="Use Brainfloat")
+        gr.Checkbox(value=False, label="Use Brainfloat"),
+        gr.Checkbox(value=False, label="Use eval()")
     ],
     outputs=gr.Audio(type="filepath", label="Generated Audio"),
     title="Stable Audio Generator",