Upload 4 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
+mobilenet_best_model.keras filter=lfs diff=lfs merge=lfs -text
+resnet_best_model.keras filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,47 @@
+import gradio as gr
+from PIL import Image
+import numpy as np
+from tensorflow.keras.preprocessing import image as keras_image
+from tensorflow.keras.applications.resnet50 import preprocess_input as resnet_preprocess_input
+from tensorflow.keras.applications.mobilenet_v2 import preprocess_input as mobilenet_preprocess_input
+from tensorflow.keras.models import load_model
+
+# Load your trained models
+resnet_model = load_model('/path/to/resnet_model.h5')  # Update path
+mobilenet_model = load_model('/path/to/mobilenet_model.h5')  # Update path
+
+def predict_comic_character(img, model_type):
+    img = Image.fromarray(img.astype('uint8'), 'RGB')
+    img = img.resize((224, 224))  # Resize the image to fit model input
+    img_array = keras_image.img_to_array(img)
+    img_array = np.expand_dims(img_array, axis=0)
+
+    if model_type == 'ResNet50':
+        img_array = resnet_preprocess_input(img_array)
+        prediction = resnet_model.predict(img_array)
+    elif model_type == 'MobileNetV2':
+        img_array = mobilenet_preprocess_input(img_array)
+        prediction = mobilenet_model.predict(img_array)
+    else:
+        return {"error": "Invalid model type selected"}
+
+    classes = ['Superman', 'Batman', 'WonderWoman', 'Riddler', 'Spider-Man', 'Iron-Man', 
+               'Hulk', 'The Joker', 'Magneto', 'Wolverine', 'Deadpool', 'Catwoman']
+    
+    return {classes[i]: float(prediction[0][i]) for i in range(len(classes))}
+
+# Define the Gradio interface
+interface = gr.Interface(
+    fn=predict_comic_character,
+    inputs=[
+        gr.inputs.Image(type="numpy", label="Upload an image of a comic character"),
+        gr.inputs.Radio(['ResNet50', 'MobileNetV2'], label="Choose Model")
+    ],
+    outputs=gr.outputs.Label(num_top_classes=5, label="Prediction"),
+    title="Comic Character Classifier",
+    description="Upload an image of a comic character and the classifier will predict the character.",
+    theme="huggingface"  # Optional: Adds a nice theme from Gradio's gallery
+)
+
+# Launch the interface
+interface.launch()

mobilenet_best_model.keras

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

requirements.txt

@@ -0,0 +1,19 @@
+blinker==1.7.0
+click==8.1.7
+Flask==3.0.2
+Flask-Cors==4.0.0
+itsdangerous==2.1.2
+Jinja2==3.1.3
+joblib==1.3.2
+MarkupSafe==2.1.5
+numpy==1.26.4
+pandas==2.2.1
+python-dateutil==2.8.2
+pytz==2024.1
+scikit-learn==1.4.1.post1
+scipy==1.12.0
+six==1.16.0
+threadpoolctl==3.3.0
+tzdata==2024.1
+Werkzeug==3.0.1
+tensorflow==2.16.1

resnet_best_model.keras

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