innat commited on
Commit
047baaf
·
1 Parent(s): bbc8456

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -32,6 +32,7 @@ LABEL_MAPS = {
32
  }
33
 
34
  def tube_mask_generator(mask_ratio):
 
35
  window_size = (
36
  num_frames // 2,
37
  input_size // patch_size[0],
@@ -49,6 +50,7 @@ def tube_mask_generator(mask_ratio):
49
 
50
 
51
  def get_model(data_type):
 
52
  ft_model = keras.models.load_model(MODELS[data_type][0])
53
  pt_model = keras.models.load_model(MODELS[data_type][1])
54
 
 
32
  }
33
 
34
  def tube_mask_generator(mask_ratio):
35
+ mask_ratio=0.8
36
  window_size = (
37
  num_frames // 2,
38
  input_size // patch_size[0],
 
50
 
51
 
52
  def get_model(data_type):
53
+ data_type = 'K400'
54
  ft_model = keras.models.load_model(MODELS[data_type][0])
55
  pt_model = keras.models.load_model(MODELS[data_type][1])
56