Update app.py

app.py

@@ -10,25 +10,18 @@ import os
 import accelerate
 
 
-
-
-
-
 # preprocess and crop audio file
-def audio_preprocess(file_name = '/test1/vocals.wav'):
+def audio_preprocess(input_file):
    # separate music and vocal
    separator = Separator('spleeter:2stems')
-   separator.separate_to_file(input_file, output_file)
-
+   #separator.separate_to_file(input_file, output_file)
+    separated_audio = separator.separate(input_file)
 
    # Crop the audio
    start_time = 60000  # e.g. 30 seconds, 30000
    end_time = 110000  # e.g. 40 seconds, 40000
 
-
-
-
-   audio = AudioSegment.from_file(file_name)
+   audio = AudioSegment.from_file(separated_audio)
    cropped_audio = audio[start_time:end_time]
    processed_audio = cropped_audio
    # .export('cropped_vocals.wav', format='wav') # save vocal audio file
@@ -79,12 +72,8 @@ def senti_model(transcription):
    final_result = pipe(transcription)
    display = f"Sentiment Analysis shows that this song is {final_result[0]['label']}. Confident level of this analysis is {final_result[0]['score']*100:.1f}%."
    print(display)
-   return display
-
-
-   # return final_result
-
 
+   return display
 
 
 # main