From f15c829927fbda3a28d4a68feca87584410a9370 Mon Sep 17 00:00:00 2001
From: Wok <woctezuma@users.noreply.github.com>
Date: Sun, 23 Aug 2020 00:20:39 +0200
Subject: [PATCH] Fix issue with MobileNetV3 in Keras

https://github.com/tensorflow/tensorflow/issues/40217
https://github.com/keras-team/keras-applications/issues/183
---
 model_utils.py   | 4 ++--
 requirements.txt | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/model_utils.py b/model_utils.py
index bd70557..317f23d 100644
--- a/model_utils.py
+++ b/model_utils.py
@@ -1,6 +1,6 @@
 import numpy as np
-from keras.applications.mobilenet_v3 import MobileNetV3Small
-from keras.applications.mobilenet_v3 import preprocess_input
+from tensorflow.keras.applications import MobileNetV3Small
+from tensorflow.keras.applications.mobilenet_v3 import preprocess_input
 from keras.preprocessing.image import img_to_array
 
 
diff --git a/requirements.txt b/requirements.txt
index e3f2a62..88a1202 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+tf-nightly
 cython==0.29.21
 numpy==1.19.1
 keras==2.4.3