Update app.py

app.py

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