Update audio2text/init.py

audio2text/init.py

@@ -1,8 +1,13 @@
-from transformers import WhisperProcessor, WhisperForConditionalGeneration
+from transformers import pipeline
 
 # ArticMonkey:19.03.24:1700 example of version name in plaintext  will be convert into hex using this site -> https://magictool.ai/tool/text-to-hex-converter/
 # Here ArticMonkey is name of version and rest of all is data and time
 
+device = 0 if torch.cuda.is_available() else "cpu"
+
 checkpoint = "openai/whisper-base"
-processor = WhisperProcessor.from_pretrained(checkpoint)
-model = WhisperForConditionalGeneration.from_pretrained(checkpoint)
+pipe = pipeline(
+    task="automatic-speech-recognition",
+    model=checkpoint,
+    device=device,
+)