Update app.py

app.py

@@ -16,11 +16,11 @@ def get_training_info(audio_file):
     """Determine training parameters based on the duration of the audio file."""
     try:
         if audio_file is None:
-            raise ValueError as e:
-            return gr.Error('Please provide an audio file.')
+            return "Please provide an audio file."
         
         duration = get_audio_duration(audio_file)
-        sample_rate = wave.open(audio_file, 'rb').getframerate()
+        with wave.open(audio_file, 'rb') as audio:
+            sample_rate = audio.getframerate()
 
         training_info = {
             (0, 2): (150, 'TITAN'),
@@ -38,7 +38,7 @@ def get_training_info(audio_file):
         
         return 'Duration is not within the specified range.'
     except ValueError as e:
-        return gr.Error(str(e))
+        return str(e)
 
 with gr.Blocks(theme=gr.themes.Base(primary_hue="sky", secondary_hue="blue"), title="RVC TRAINING HELPER") as demo:
     with gr.Tab("Main Settings"):