Update run.py

run.py

@@ -1,8 +1,12 @@
 import gradio as gr
-from huggingface.user import HFUser, GR_CONF
+from user import HFUser, GR_CONF
 
 Theme = gr.Theme.load(GR_CONF["theme"])
 GR_CONF["theme"] = Theme
+dataset_options = [
+    "OpenVideo/pexels-raw",
+    "OpenVideo/Sample-2k",
+]
 
 def login(token):
     u = HFUser.from_token(token)
@@ -12,6 +16,7 @@ def show_time(u, name):
     return u.ping(name), gr.Column(visible=True)
 
 def list_dataset(u, repo):
+    print(f"repo is: {repo}")
     files = u.list_dataset(repo)
     if repo == "OpenVideo/Sample-2k":
         print(f"files is: {files}")
@@ -57,7 +62,8 @@ with gr.Blocks(**GR_CONF) as Core:
                 time= gr.Textbox(label="Time", interactive=False, scale=1)
 
             with gr.Column(visible=False) as Repo:
-                raw_dataset= gr.Textbox("OpenVideo/pexels-raw", label="Raw Dataset")
+                # raw_dataset= gr.Textbox("OpenVideo/pexels-raw", label="Raw Dataset")
+                raw_dataset = gr.Dropdown(choices=dataset_options, label="Raw Dataset")
                 _list = gr.Button("List", variant='secondary', size='sm')
 
             with gr.Column(visible=False) as Batch: