Update app.py

app.py

@@ -116,6 +116,7 @@ async def tts(ws: WebSocket):
                 logits_processor= [masker],
                 do_sample=True, temperature=0.7, top_p=0.95,
                 use_cache=True,
+                return_dict_in_generate=True, # Added return_dict_in_generate
                 return_legacy_cache=True # Added legacy cache
             )