Initial Commit

app.py

@@ -0,0 +1,40 @@
+import gradio as gr
+import numpy as np
+from tensorflow.keras.models import load_model
+from PIL import Image
+
+model = load_model("final/brain_tumor_fine_tune.h5")
+model.load_weights("final/brain_tumor_fine_tune.h5")
+
+
+class_names = [
+    'glioma',
+    'meningioma',
+    'notumor',
+    'pituitary',
+]
+
+def classify_tumor(input_image):
+    
+    img = Image.fromarray(input_image)
+    resized_img = img.resize((299, 299))
+    img = np.asarray(resized_img)
+    img = np.expand_dims(img, axis=0)
+    img = img / 255
+    prediction = model.predict(img)
+    
+    confidences = {class_names[i]: float(prediction[0][i]) for i in range(len(class_names))}
+    
+    return {class_name: conf for class_name, conf in sorted(confidences.items(), key=lambda x: x[1], reverse=True)}
+
+iface = gr.Interface(
+    fn=classify_tumor,
+    inputs=gr.Image(type="numpy"),
+    outputs=gr.Label(num_top_classes=5),
+    title="Brain Tumor Classification",
+    description="Upload an image of a Brain to classify its tumor.",
+    theme=gr.themes.Soft()
+)
+
+# Launch the interface
+iface.launch(share=True)

final/brain_tumor_fine_tune.h5

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