Sa-m commited on
Commit
1956924
1 Parent(s): cad374a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -4,8 +4,11 @@ import tensorflow as tf
4
  import numpy as np
5
  from tensorflow.keras.models import load_model
6
  import os
7
- os.environ['TF_KERAS'] = '1'
8
- os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
 
 
 
9
  model=load_model('best_model_76.h5')
10
 
11
  def classify_image(inp):
 
4
  import numpy as np
5
  from tensorflow.keras.models import load_model
6
  import os
7
+
8
+ tf.compat.v1.disable_v2_behavior()
9
+
10
+ #os.environ['TF_KERAS'] = '1'
11
+ #os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
12
  model=load_model('best_model_76.h5')
13
 
14
  def classify_image(inp):