Update app.py

app.py

@@ -6,8 +6,8 @@ from keras.models import load_model
 import tensorflow as tf
 from tensorflow import keras
 
-# Cargar el modelo entrenado
-model1 = load_model('./isatron_v3.h5')
+# Cargar el nuevo modelo entrenado
+model1 = load_model('./modelo_isatron_jeysshonl.h5')
 
 # Función para encontrar la última capa convolucional
 def find_last_conv_layer(model):
@@ -100,14 +100,15 @@ def image_classifier1(img):
     prediction_percentage = {'PNEUMONIA': float(prediction), 'NORMAL': float(1 - prediction)}
     return overlayed_img, prediction_percentage
 
-# Definir CSS personalizado
+# Definir CSS personalizado con fondo oscuro
 custom_css = """
 body {
-    background-color: #e6f2ff;
+    background-color: #1a1a1a;
+    color: #e6e6e6;
 }
 
 h1 {
-    color: #003366;
+    color: #ffcc00;
     text-align: center;
 }
 
@@ -116,24 +117,25 @@ h1 {
     max-width: 800px;
     margin: auto;
     padding: 20px;
-    background-color: #ffffff;
+    background-color: #333333;
     border-radius: 10px;
+    border: 1px solid #666666;
 }
 
 .footer {
     text-align: center;
     margin-top: 20px;
-    color: #003366;
+    color: #e6e6e6;
     font-size: 14px;
 }
 
 label {
     font-weight: bold;
-    color: #003366;
+    color: #e6e6e6;
 }
 
 button {
-    background-color: #0059b3;
+    background-color: #ff6600;
     color: white;
     border: none;
     padding: 10px 20px;
@@ -141,7 +143,7 @@ button {
 }
 
 button:hover {
-    background-color: #004080;
+    background-color: #cc5200;
 }
 """
 
@@ -153,7 +155,7 @@ demo_model1 = gr.Interface(
         gr.Image(type="numpy", label="Imagen con Mapa de Calor"),
         gr.Label(num_top_classes=2, label="Resultados de Predicción")
     ],
-    title="IsaTron V2 con Mapa de Calor",
+    title="IsaTron V2 ",
     description="Esta aplicación detecta neumonía a partir de imágenes de radiografías de tórax.",
     article="<div class='footer'>Jeysshon 2024</div>",
     css=custom_css