Update app.py

app.py

@@ -11,7 +11,7 @@ def get_audio_duration(audio_file):
 
 def get_training_info(audio_file):
     if audio_file is None:
-        return 'the audio file pls'
+        gr.Error('the audio file pls')
     
     duration = get_audio_duration(audio_file)
     sample_rate = wave.open(audio_file, 'rb').getframerate()
@@ -30,7 +30,7 @@ def get_training_info(audio_file):
         if min_duration <= duration < max_duration:
             break
     else:
-        return 'Duration is not within the specified range.'
+        gr.Info('Duration is not within the specified range.'
 
     return f'You should use the **{pretrain}** pretrain with **{epochs}** epochs at **{sample_rate/1000}kHz** sample rate. Good luck with your training!'