diff --git a/README.md b/README.md
index 0fb7ac8..a2f259b 100644
--- a/README.md
+++ b/README.md
@@ -7,6 +7,7 @@ Transcriber code
 ## to-do
 
 - seperate data production form code
+- seperate huggingface (hf) token out
 
 ```
 project_root/
diff --git a/requirements.txt b/requirements.txt
index 2300225..8ed7f28 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 youtube-dl
-pfy
 pytube
 pydub
+torch
 git+https://github.com/m-bain/whisperx.git
\ No newline at end of file
diff --git a/src/main.py b/src/main.py
index 9a3a7b0..50e0033 100644
--- a/src/main.py
+++ b/src/main.py
@@ -4,8 +4,7 @@ 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',
-        '/path/to/your/local/file.mp3'
+        'https://www.youtube.com/watch?v=BVVo1BaL6JA'
     ]
 
     transcript_generator = TranscriptGenerator()