Update run.py

run.py

@@ -10,12 +10,10 @@ def load_hf_dataset(dataset_path, auth_token):
     dataset = load_dataset(dataset_path, token=auth_token)
     
     video_paths = dataset
-    print("load done")
     
     return video_paths
 
 def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, endpoint, video_src, video_hf, video_hf_auth, video_od, video_od_auth, video_gd, video_gd_auth, frame_format, frame_limit):
-    print("begin caption")
     if video_src:
         video = video_src
         processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
@@ -31,14 +29,12 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
         caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
         return f"{caption}", f"Using model '{model}' with {len(frames)} frames extracted.", debug_image
     elif video_hf and video_hf_auth:
-        print("begin video_hf")
         # Handle Hugging Face dataset
         video_paths = load_hf_dataset(video_hf, video_hf_auth)
         video_paths = video_paths["train"]
         # Process all videos in the dataset
         all_captions = []
         for video_path_url in video_paths:
-            print("video_path")
             video_path_url = video_path_url["id"]
             # 使用requests下载文件到临时文件
             response = requests.get(video_path_url, stream=True)
@@ -117,6 +113,9 @@ with gr.Blocks() as Core:
                         with gr.Tab("HF"):
                             video_hf = gr.Text(label="Huggingface File Path")
                             video_hf_auth = gr.Text(label="Huggingface Token")
+                        with gr.Tab("Parquet_index"):
+                            video_hf = gr.Text(label="Parquet_index")
+                            video_hf_auth = gr.Text(label="Huggingface Token")
                         with gr.Tab("Onedrive"):
                             video_od = gr.Text("Microsoft Onedrive")
                             video_od_auth = gr.Text(label="Microsoft Onedrive Token")