Update app.py

app.py

@@ -8,7 +8,7 @@ from tensorflow.keras.applications.convnext import preprocess_input # type: igno
 import gradio as gr
 
 # Load the model
-model = load_model('models/ConvNeXtBase_80_tresh_spp.tf')
+model = keras.layers.TFSMLayer('models/ConvNeXtBase_80_tresh_spp.tf')
 
 # Load the taxonomy .csv
 taxo_df = pd.read_csv('taxonomy/taxonomy_mapping.csv', sep=';')