Update app.py

app.py

@@ -5,12 +5,12 @@ from load_dataset import classes
 
 nm_model = tf.keras.models.load_model("mn_model.keras")
 
-# resnet_model = tf.keras.models.load_model("../models/newmodel.h5")
+resnet_model = tf.keras.models.load_model("resnet_best.h5")
 
 inception_model = tf.keras.models.load_model("inception_v3.keras")
 
 cifar10_labels = classes
-models = [ "InceptionBased Model", "MobileNetBased Model"]
+models = [ "InceptionBased Model", "MobileNetBased Model", "ResNetBased Model"]
 
 
 def classify_image(input_image, model_name):
@@ -30,8 +30,8 @@ def classify_image(input_image, model_name):
 def get_model(model_name):
     if model_name == "MobileNetBased Model":
         return nm_model
-    # elif model_name == "ResNetBased Model":
-    #     return resnet_model
+    elif model_name == "ResNetBased Model":
+        return resnet_model
     elif model_name == "InceptionBased Model":
         return inception_model