update

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 
+from tensorflow.keras import models 
+from tensorflow.keras.preprocessing.image import load_img
 import tensorflow as tf
 from hugsvision.inference.TorchVisionClassifierInference import TorchVisionClassifierInference
 
@@ -33,7 +35,7 @@ def predict_image(image, model_name):
         image = np.array(image) / 255
         image = np.expand_dims(image, axis=0)
 
-        model = "./models/" + model_name + "model.h5"
+        model = model = models.load_model("./models/" + model_name + "/model.h5")
         pred = model.predict(image)
 
         pred = dict((labels[i], "%.2f" % pred[0][i]) for i in range(len(labels)))