Update app.py

app.py

@@ -1,9 +1,10 @@
 import fastapi
 from transformers import pipeline
 import h5py
+import keras
 
-# Load the model from the h5 file
-model = h5py.File("model_finetuned.h5", "r")["model"]
+# Load the model from the HDF5 file
+model = keras.models.load_model("model_finetuned.h5")]
 
 # Define a function to preprocess the image
 def preprocess_image(image):