From 767b2028b57648ebe9fbbd3cd3357a98a74d3e9b Mon Sep 17 00:00:00 2001
From: serengil <sefik.serengil@softtech.com.tr>
Date: Sun, 13 Dec 2020 20:10:41 +0300
Subject: [PATCH] tf2

---
 deepface/basemodels/VGGFace.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/deepface/basemodels/VGGFace.py b/deepface/basemodels/VGGFace.py
index 3ba4fa1..4870593 100644
--- a/deepface/basemodels/VGGFace.py
+++ b/deepface/basemodels/VGGFace.py
@@ -3,8 +3,9 @@ from pathlib import Path
 import gdown
 
 import tensorflow as tf
+tf_version = int(tf.__version__.split(".")[0])
 
-if int(tf.__version__.split(".")[0]) == 1:
+if tf_version == 1:
 	from keras.models import Model, Sequential
 	from keras.layers import Input, Convolution2D, ZeroPadding2D, MaxPooling2D, Flatten, Dense, Dropout, Activation
 else: