Update main.py
Browse files
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'] = './
|
| 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 |
-
|
| 619 |
-
|
| 620 |
-
|
| 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=[
|