Update run.py

run.py

@@ -27,20 +27,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
             writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
             writer.writeheader()
     
-            if video_src:
-                video = video_src
-                processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
-                frames = processor._decode(video)
-                base64_list = processor.to_base64_list(frames)
-                debug_image = processor.concatenate(frames)
-                if not key or not endpoint:
-                    return "", f"API key or endpoint is missing. Processed {len(frames)} frames.", debug_image
-                api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
-                caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
-                progress_info.append(f"Using model '{model}' with {len(frames)} frames extracted.")
-                writer.writerow({'md5': 'single_video', 'caption': caption})
-                return f"{caption}", "\n".join(progress_info), debug_image
-            elif video_hf and video_hf_auth:
+            if video_hf and video_hf_auth:
                 progress_info.append('Begin processing Hugging Face dataset.')
                 temp_parquet_file = hf_hub_download(
                     repo_id=video_hf,
@@ -118,8 +105,6 @@ with gr.Blocks() as Core:
                             endpoint_oai = gr.Textbox(label="OpenAI Endpoint")
                 with gr.Accordion("Data Source", open=True):
                     with gr.Tabs():
-                        with gr.Tab("Upload"):
-                            video_src = gr.Video(sources="upload", show_label=False, show_share_button=False, mirror_webcam=False)
                         with gr.Tab("HF"):
                             video_hf = gr.Text(label="Huggingface File Path")
                             video_hf_auth = gr.Text(label="Huggingface Token")