Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-os.system('pip uninstall vidfetch')
+os.system('pip uninstall vidfetch -y')
 os.system('pip install -U https://github.com/dailingx/VidFetch/archive/master.zip')
 os.system('pip install --upgrade google-api-python-client')
 
@@ -10,13 +10,17 @@ from vidfetch.website.youtube import YoutubeVideoDataset
 
 def fetch(
     key_word: str,
-    dev_key: str
+    dev_key: str,
+    hf_token: str,
+    hf_ds_repo_id: str
 ):
     youtube_video_dataset = YoutubeVideoDataset(
         root_dir="./",
         google_cloud_developer_key=dev_key,
         search_keyword=key_word,
-        video_max_num=1
+        video_max_num=1,
+        hf_token=hf_token,
+        hf_ds_repo_id=hf_ds_repo_id
     )
     youtube_video_dataset.download()
     return key_word
@@ -28,10 +32,13 @@ with gr.Blocks() as demo:
         with gr.Column():
             kw_input_text = gr.Text(label='Keyword')
             dev_key_input_text = gr.Text(label='Google Cloud Developer Key')
+            hf_token_input_text = gr.Text(label='HF Token')
+            hf_ds_repo_id_text = gr.Text(label='HF Dataset Repo ID, like: OpenVideo/YouTube-Commons-5G-Raw')
             fetch_btn = gr.Button("Fetch")
         result = gr.Text()
 
-    fetch_btn.click(fn=fetch, inputs=[kw_input_text, dev_key_input_text], outputs=[result])
+    fetch_btn.click(fn=fetch, inputs=[kw_input_text, dev_key_input_text, hf_token_input_text, hf_ds_repo_id_text],
+                    outputs=[result])
 
 
 if __name__ == "__main__":