innat commited on
Commit
4640262
1 Parent(s): 3a82e83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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)