Upload app.py

app.py

@@ -2,11 +2,10 @@ import gradio as gr
 import tensorflow as tf
 from tensorflow.keras.applications.inception_resnet_v2 import preprocess_input
 from tensorflow.keras.preprocessing import image
-from tensorflow.keras.models import load_model
 import numpy as np
 from PIL import Image
 
-model = load_model('best_model_InceptionV2.h5')
+model = tf.keras.models.load_model('best_model_InceptionV2.keras')
 
 # Function for prediction
 def predict(img):