Upload 3 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+inception_v3.keras filter=lfs diff=lfs merge=lfs -text
+mn_model.keras filter=lfs diff=lfs merge=lfs -text

inception_v3.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f2d751d2d613deb54f2b8f8b97cff8eecd7c0f87ed7a548e8028af68b9024ab6
+size 298791432

index.py

@@ -0,0 +1,46 @@
+import tensorflow as tf
+import gradio as gr
+
+from src.load_dataset.load_dataset import classes
+
+nm_model = tf.keras.models.load_model("../models/mn_model.keras")
+
+resnet_model = tf.keras.models.load_model("../models/newmodel.h5")
+
+inception_model = tf.keras.models.load_model("../models/inception_v3.keras")
+
+cifar10_labels = classes
+models = ["ResNetBased Model", "MobileNetBased Model", "InceptionBased Model"]
+
+
+def classify_image(input_image, model_name):
+    try:
+        input_image = tf.image.resize(input_image, (32, 32))
+        labels = cifar10_labels
+        model = get_model(model_name)
+        input_image = tf.expand_dims(input_image, axis=0)
+        predictions = model.predict(input_image).flatten()
+        top_indices = predictions.argsort()[-10:][::-1]
+        confidences = {labels[i]: float(predictions[i]) for i in top_indices}
+        return confidences
+    except Exception as e:
+        return {"error": str(e)}
+
+
+def get_model(model_name):
+    if model_name == "MobileNetBased Model":
+        return nm_model
+    elif model_name == "ResNetBased Model":
+        return resnet_model
+    elif model_name == "InceptionBased Model":
+        return inception_model
+
+
+interface = gr.Interface(
+    fn=classify_image,
+    inputs=[gr.Image(type="numpy", image_mode="RGB", label="Input Image"),
+            gr.Dropdown(models, label="Model Choice")],
+    outputs=gr.Label(num_top_classes=3, label="Predictions"),
+)
+
+interface.launch(debug=False, share=True)

mn_model.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d1da46963093525e7af3ea4afc97eb2fe4f30144812cf8c4cac2fe998d03976
+size 73589384