Update app.py

app.py

@@ -5,6 +5,7 @@ os.system("pip install https://github.com/kpu/kenlm/archive/master.zip")
 os.system("pip install pyctcdecode")
 os.system("pip install gradio")
 os.system("pip install librosa")
+os.system("pip install torch")
 
 import gradio as gr
 import librosa
@@ -19,6 +20,7 @@ from transformers import Wav2Vec2ProcessorWithLM
 repo_name = "aiface/vietnamese_s2t"
 
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+# device="cpu"
 processor = Wav2Vec2ProcessorWithLM.from_pretrained(repo_name, token="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")
 model = Wav2Vec2ForCTC.from_pretrained(repo_name, token="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").to(device)
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(repo_name, token="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")