Added poetry, improve project structure

1 files changed, 12 insertions, 0 deletions

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())