Update app.py

app.py

@@ -17,8 +17,8 @@ from transformers import (
     Trainer,
     pipeline
 )
-# processor = Wav2Vec2Processor.from_pretrained("airesearch/wav2vec2-large-xlsr-53-th")
-# model = Wav2Vec2ForCTC.from_pretrained("BALAKA/wav2vec2-large-xlsr-53-thai")
+processor = Wav2Vec2Processor.from_pretrained("airesearch/wav2vec2-large-xlsr-53-th")
+model = Wav2Vec2ForCTC.from_pretrained("BALAKA/wav2vec2-large-xlsr-53-thai")
 
 demo = gr.Blocks()
 
@@ -50,7 +50,7 @@ def tran_script(file_path):
         predicted_sentence = processor.batch_decode(predicted_ids)
         return predicted_sentence
     else:
-        now_path = glob('/content/split_*.mp3')
+        now_path = glob('/home/user/app/split_*.mp3')
         sentence = []
         for i in range(file_path - 1):
             now_path = f'/content/split_{i+1}.mp3'
@@ -115,14 +115,13 @@ def youtube_loader(link):
     yt = YouTube(str(link))
     video = yt.streams.filter(only_audio=True).first()
     out_file = video.download(output_path='mp3')
-    return out_file
     os.rename(out_file, 'youtube.mp3')
-    return process('/content/mp3/youtube.mp3')
+    return process('/home/user/app/mp3/youtube.mp3')
 
 
 def twitch_loader(link):
     os.system(f"twitch-dl download -q audio_only {link} --output twitch.wav")
-    return process('/content/twitch.wav')
+    return process('/home/user/app/twitch.wav')
 
 
 with demo: