Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 import plotly.graph_objects as go
 
 # Load your trained CIFAR model
-model = lm('build_model_1_v2.keras')
+model = lm('build_model_1_v2.h5')
 
 # Define the CIFAR-10 class names
 class_names = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']