aus10powell commited on
Commit
40b2e47
·
1 Parent(s): 23c48ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import io, os
5
  import logging
6
  import keras_metrics
7
  from tensorflow import keras
 
8
  import utils
9
  ## Configs
10
  keras.utils.get_custom_objects()['recall'] = utils.recall
 
5
  import logging
6
  import keras_metrics
7
  from tensorflow import keras
8
+ from tensorflow.python.compiler.tensorrt import trt_convert as trt
9
  import utils
10
  ## Configs
11
  keras.utils.get_custom_objects()['recall'] = utils.recall