Update app.py

app.py

@@ -46,6 +46,10 @@ class AudioMask(LogitsProcessor):
             allowed_audio
         ])
 
+        # Penalize NEW_BLOCK if buffer is not empty
+        if self.buffer_pos > 0:
+            logits[:, NEW_BLOCK] = float("-inf") # Apply a large negative penalty
+
         if self.sent_blocks:                        # ab 1. Block EOS zulassen
             allowed = torch.cat([allowed, self.eos])