Update app.py

app.py

@@ -1,7 +1,7 @@
 import tensorflow as tf
-from tensorflow.keras.applications.resnet_v2 import ResNet50V2
 from tensorflow.keras.preprocessing import image
-from tensorflow.keras.applications.resnet_v2 import preprocess_input, decode_predictions
+from tensorflow.keras.applications.mobilenet_v2 import MobileNetV2 as keras_model
+from tensorflow.keras.applications.mobilenet_v2 import preprocess_input, decode_predictions
 import matplotlib.pyplot as plt
 from alibi.explainers import IntegratedGradients
 from alibi.datasets import load_cats
@@ -26,7 +26,7 @@ url = "https://upload.wikimedia.org/wikipedia/commons/4/43/Cute_dog.jpg"
 path_input = "./dog.jpg"
 urllib.request.urlretrieve(url, filename=path_input)
 
-model = ResNet50V2(weights='imagenet')
+model = keras_model(weights='imagenet')
 
 n_steps = 50
 method = "gausslegendre"