Update app.py
Browse files
app.py
CHANGED
@@ -36,6 +36,12 @@ def get_model(data_type):
|
|
36 |
|
37 |
|
38 |
def inference(video_file, dataset_type, mask_ratio):
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
container = read_video(video_file)
|
40 |
frames = frame_sampling(container, num_frames=num_frames)
|
41 |
bool_masked_pos_tf = tube_mask_generator(mask_ratio)
|
|
|
36 |
|
37 |
|
38 |
def inference(video_file, dataset_type, mask_ratio):
|
39 |
+
print('---------------------------')
|
40 |
+
print(video_file)
|
41 |
+
print(dataset_type)
|
42 |
+
print(mask_ratio)
|
43 |
+
print('---------------------------')
|
44 |
+
|
45 |
container = read_video(video_file)
|
46 |
frames = frame_sampling(container, num_frames=num_frames)
|
47 |
bool_masked_pos_tf = tube_mask_generator(mask_ratio)
|