Kimata commited on
Commit
8fa56f0
·
1 Parent(s): 2c3194b

enable ONEDNN ops

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. pipeline.py +1 -0
app.py CHANGED
@@ -29,7 +29,7 @@ audio_interface = gr.Interface(pipeline.deepfakes_audio_predict,
29
 
30
 
31
  app = gr.TabbedInterface(interface_list= [image_interface, video_interface, audio_interface],
32
- tab_names = ['Image inference', 'Video inference', 'audio_interface'])
33
 
34
  if __name__ == '__main__':
35
  app.launch(share = False)
 
29
 
30
 
31
  app = gr.TabbedInterface(interface_list= [image_interface, video_interface, audio_interface],
32
+ tab_names = ['Image inference', 'Video inference', 'Audio inference'])
33
 
34
  if __name__ == '__main__':
35
  app.launch(share = False)
pipeline.py CHANGED
@@ -7,6 +7,7 @@ import numpy as np
7
  import tensorflow as tf
8
  from facenet_pytorch import MTCNN
9
  from rawnet import RawNet
 
10
 
11
  local_zip = "./efficientnet-b0.zip"
12
  zip_ref = zipfile.ZipFile(local_zip, 'r')
 
7
  import tensorflow as tf
8
  from facenet_pytorch import MTCNN
9
  from rawnet import RawNet
10
+ os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
11
 
12
  local_zip = "./efficientnet-b0.zip"
13
  zip_ref = zipfile.ZipFile(local_zip, 'r')