Update app.py

app.py

@@ -3,14 +3,14 @@ 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")
 
 inception_model = tf.keras.models.load_model("inception_v3.keras")
 
 cifar10_labels = classes
-models = [ "MobileNetBased Model", "InceptionBased Model"]
+models = [ "InceptionBased Model"]
 
 
 def classify_image(input_image, model_name):
@@ -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(