Update app.py

app.py

@@ -28,6 +28,7 @@ def audio_preprocess(file_name = '/test1/vocals.wav'):
     # .export('cropped_vocals.wav', format='wav') # save vocal audio file
     return processed_audio
 
+
 # ASR transcription
 def asr_model(processed_audio):
     # load audio file
@@ -68,12 +69,13 @@ def senti_model(transcription):
 
 
 # main
-def main():
+def main(input_file):
 
-    # processed_audio = audio_preprocess(input_file)
-    processed_audio = input_file
+    processed_audio = audio_preprocess(input_file)
+    # processed_audio = input_file
 
     transcription = asr_model(processed_audio)
+    st.write(transcription)
     final_result = senti_model(transcription)
     st.write(final_result)
 
@@ -99,7 +101,7 @@ if __name__ == '__main__':
 
     # load song
     #input_file = os.path.isfile("test1.mp3")
-    output_file = os.path.isdir("")
+    # output_file = os.path.isdir("")
 
     if button_click:
-        main()
+        main(input_file=input_file)