Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 
 from load_dataset import classes
 
-# nm_model = tf.keras.models.load_model("mn_model.keras")
+nm_model = tf.keras.models.load_model("mn_model.keras")
 
 # resnet_model = tf.keras.models.load_model("../models/newmodel.h5")
 
@@ -28,12 +28,12 @@ def classify_image(input_image, model_name):
 
 
 def get_model(model_name):
-    # if model_name == "MobileNetBased Model":
-    #     return nm_model
+    if model_name == "MobileNetBased Model":
+        return nm_model
     # elif model_name == "ResNetBased Model":
     #     return resnet_model
-    # elif model_name == "InceptionBased Model":
-    return inception_model
+    elif model_name == "InceptionBased Model":
+        return inception_model
 
 
 interface = gr.Interface(