add flags

app.py

@@ -15,6 +15,7 @@ import cv2
 from tqdm import tqdm
 
 
+TF_ENABLE_ONEDNN_OPTS=0
 class Sampling(tf.keras.layers.Layer):
     def call(self, inputs):
         mean, log_var = inputs
@@ -24,7 +25,6 @@ class Sampling(tf.keras.layers.Layer):
 custom_objects = {'Sampling': Sampling}
 
 
-
 '''
 
 Pre-trained models - Instructions about training you can find in another file