From cc401846811207b8aa2da881f52e8d37f06ef6cd Mon Sep 17 00:00:00 2001
From: Pherkel
Date: Thu, 17 Aug 2023 18:53:35 +0200
Subject: Added poetry, improve project structure

---
 swr2_asr/__main__.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 swr2_asr/__main__.py

(limited to 'swr2_asr/__main__.py')

diff --git a/swr2_asr/__main__.py b/swr2_asr/__main__.py
new file mode 100644
index 0000000..be294fb
--- /dev/null
+++ b/swr2_asr/__main__.py
@@ -0,0 +1,12 @@
+"""Main entrypoint for swr2-asr."""
+import torch
+import torchaudio
+
+if __name__ == "__main__":
+    # test if GPU is available
+    print("GPU available: ", torch.cuda.is_available())
+
+    # test if torchaudio is installed correctly
+    print("torchaudio version: ", torchaudio.__version__)
+    print("torchaudio backend: ", torchaudio.get_audio_backend())
+    print("torchaudio info: ", torchaudio.get_audio_backend())
-- 
cgit v1.2.3