Update app.py

app.py

@@ -105,6 +105,16 @@ def process_fn(inp_audio, option_text):
     if option_text == "Generate Watermark":
         with torch.no_grad():
             wm_sig = embed(sig.clone(), generator).normalize(orig_loud).ensure_max_of_audio()
+            output_labels.labels.append(
+                AudioLabel(
+                    t = 0,
+                    label = "watermark: 1.0",
+                    duration = wm_sig.duration,
+                    description = f"watermark confidence: 1.0, start: 0.0s, end: {wm_sig.duration:.2f}s",
+                    color = OutputLabel.rgb_color_to_int(255, 0, 0),
+                    amplitude = 1.0
+                )
+            )
             return save_audio(wm_sig), output_labels
     else:
         with torch.no_grad():