Update app.py

app.py

@@ -58,6 +58,7 @@ def predict(prompt, style, audio_file_pth, agree):
                 None,
             )
         style = 'en_' + style
+        prompt_length = len(prompt.split(' '))
 
     else:
         if style not in ['default']:
@@ -69,10 +70,11 @@ def predict(prompt, style, audio_file_pth, agree):
                 None,
             )
         style = 'cn_' + style
+        prompt_length = len(prompt)
 
     speaker_wav = audio_file_pth
 
-    if len(prompt) < 2:
+    if prompt_length < 2:
         text_hint += f"[ERROR] Please give a longer prompt text \n"
         gr.Warning("Please give a longer prompt text")
         return (
@@ -80,8 +82,8 @@ def predict(prompt, style, audio_file_pth, agree):
             None,
             None,
         )
-    if len(prompt) > 200:
-        text_hint += f"[ERROR] Text length limited to 200 characters for this demo, please try shorter text. You can clone our open-source repo and try for your usage \n"
+    if prompt_length > 50:
+        text_hint += f"[ERROR] Text length limited to 50 words for this demo, please try shorter text. You can clone our open-source repo and try for your usage \n"
         gr.Warning(
             "Text length limited to 200 characters for this demo, please try shorter text. You can clone our open-source repo for your usage"
         )