migrate to tenforflowf 2.0

main.py

@@ -4,7 +4,8 @@ import argparse
 from glob import glob
 
 from PIL import Image
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
+tf.disable_v2_behavior()
 
 from model import lowlight_enhance
 from utils import *

model.py

@@ -5,7 +5,9 @@ import time
 import random
 
 from PIL import Image
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
+tf.disable_v2_behavior()
+
 import numpy as np
 
 from utils import *