lungDisease

README.md

@@ -1,12 +1,14 @@
 ---
-title: LungDisease
-emoji: 🔥
-colorFrom: pink
-colorTo: green
+title: PULMONET
+emoji: 🐢
+colorFrom: purple
+colorTo: gray
 sdk: gradio
 sdk_version: 5.23.1
 app_file: app.py
 pinned: false
+license: apache-2.0
+short_description: A model to predict lung ill on x-ray images
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,51 @@
+import gradio as gr
+import tensorflow as tf
+import numpy as np
+from PIL import Image
+
+# Load model
+print("Loading model...")
+modelo = tf.keras.models.load_model("modelo_multilabel_mejorado3.h5")
+print("Model loaded.")
+
+# Model classes
+target_classes = ['Atelectasis', 'Effusion', 'Infiltration', 'Nodule', 'Pneumothorax']
+UMBRAL_PROBABILIDAD = 0.3  # Minimum 30% to consider a disease relevant
+
+def predecir(img):
+    img = img.resize((224, 224))
+    img = np.array(img) / 255.0
+    img = np.expand_dims(img, axis=0)
+    pred = modelo.predict(img)[0]
+    resultados = [(target_classes[i], prob) for i, prob in enumerate(pred)]
+    resultados = [(enf, p) for enf, p in resultados if p >= UMBRAL_PROBABILIDAD]
+    resultados.sort(key=lambda x: x[1], reverse=True)
+    if not resultados:
+        return "ERROR "
+    salida = "\n".join([f"{enf}: {p*100:.2f}%" for enf, p in resultados[:2]])
+    return salida
+
+# Custom CSS to style the background and iframe lines
+custom_css = """
+body {
+    background-color: #efefef;
+}
+iframe {
+    border-top: 4px solid #000;
+    border-bottom: 4px solid #000;
+}
+"""
+
+# Gradio Interface
+demo = gr.Interface(
+    fn=predecir,
+    inputs=gr.Image(type="pil", label="Sube tu imagen"),
+    outputs="text",
+    title="Detection of Pulmonary Diseases",
+    description="Upload a chest X-ray, and the model will identify possible diseases",
+    theme="default",
+    css=custom_css
+)
+
+# Launch application
+demo.launch(server_name="0.0.0.0", share=True)

modelo_multilabel_mejorado.h5

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

modelo_multilabel_mejorado3.h5

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

requirements.txt

@@ -0,0 +1,4 @@
+tensorflow
+numpy
+pillow
+gradio