Test: ran the code and checked for ffmpeg functionality

main.py

@@ -0,0 +1,20 @@
+import os
+
+from src.transcript_generator import TranscriptGenerator
+
+
+def main():
+    # input could be a YouTube link or a local file
+    input_sources = ["https://www.youtube.com/watch?v=BVVo1BaL6JA"]
+
+    hf_token = os.environ.get("HF_TOKEN")
+    if not hf_token:
+        hf_token = input("Please enter your HF_TOKEN: ")
+    os.environ["HF_TOKEN"] = hf_token
+
+    transcript_generator = TranscriptGenerator()
+    transcript_generator.generate(input_sources)
+
+
+if __name__ == "__main__":
+    main()