Darknsu commited on
Commit
ace27e8
·
verified ·
1 Parent(s): 22c8c02

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -480,7 +480,7 @@ def process_input(video_file, npz_file, checkpoint_path, split_number):
480
  opt = vars(opt)
481
  opt['mode'] = 'test'
482
  opt['split'] = str(split_number)
483
- opt['checkpoint_path'] = './checkpoints'
484
  opt['video_feature_all_test'] = './data/I3D/'
485
  opt['anchors'] = [int(item) for item in opt['anchors'].split(',')]
486
  opt['batch_size'] = 1 # Single video processing
@@ -615,9 +615,9 @@ def process_input(video_file, npz_file, checkpoint_path, split_number):
615
  iface = gr.Interface(
616
  fn=process_input,
617
  inputs=[
618
- gr.Video(label="Upload Video (Optional, requires .npz for processing)"),
619
- gr.File(label="Upload I3D .npz File"),
620
- gr.File(label="Upload Model Checkpoint (.pth.tar)", file_types=[".pth.tar"]),
621
  gr.Dropdown(label="Split Number", choices=["1", "2", "3"], value="1")
622
  ],
623
  outputs=[
 
480
  opt = vars(opt)
481
  opt['mode'] = 'test'
482
  opt['split'] = str(split_number)
483
+ opt['checkpoint_path'] = './checkpoint'
484
  opt['video_feature_all_test'] = './data/I3D/'
485
  opt['anchors'] = [int(item) for item in opt['anchors'].split(',')]
486
  opt['batch_size'] = 1 # Single video processing
 
615
  iface = gr.Interface(
616
  fn=process_input,
617
  inputs=[
618
+ # gr.Video(label="Upload Video (Optional, requires .npz for processing)"),
619
+ # gr.File(label="Upload I3D .npz File"),
620
+ # gr.File(label="Upload Model Checkpoint (.pth.tar)", file_types=[".pth.tar"]),
621
  gr.Dropdown(label="Split Number", choices=["1", "2", "3"], value="1")
622
  ],
623
  outputs=[