sanzgiri commited on
Commit
b4d894b
1 Parent(s): 3a397ef

fix for tf1 compat

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,11 +1,13 @@
1
  import os
2
  import cv2
3
  import numpy as np
4
- import tensorflow as tf
 
5
  import tf_slim as slim
6
  import streamlit as st
7
  from PIL import Image
8
 
 
9
 
10
  def tf_box_filter(x, r):
11
  k_size = int(2 * r + 1)
 
1
  import os
2
  import cv2
3
  import numpy as np
4
+ #import tensorflow as tf
5
+ import tensorflow.compat.v1 as tf
6
  import tf_slim as slim
7
  import streamlit as st
8
  from PIL import Image
9
 
10
+ tf.disable_v2_behavior()
11
 
12
  def tf_box_filter(x, r):
13
  k_size = int(2 * r + 1)