innat commited on
Commit
808f251
1 Parent(s): 8adf29c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -46,6 +46,8 @@ def inference(video_file, model_type):
46
  model.trainable = False
47
 
48
  # inference on model
 
 
49
  outputs = model(frames[None, ...], training=False)
50
  probabilities = tf.nn.softmax(outputs).numpy().squeeze(0)
51
  confidences = {
 
46
  model.trainable = False
47
 
48
  # inference on model
49
+ print(model.summary())
50
+ print(frames.shape)
51
  outputs = model(frames[None, ...], training=False)
52
  probabilities = tf.nn.softmax(outputs).numpy().squeeze(0)
53
  confidences = {