Carlos012 commited on
Commit
bde43a0
·
1 Parent(s): bbd9283

feat (interfaz): Actualizar interfaz gradio

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -632,11 +632,11 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
632
  # Encabezado
633
  gr.Markdown(
634
  """
635
- # 🥔 Clasificador de Calidad de Papas - Comparación de Modelos
636
- ### Vision Transformer (ViT-Base/16) - Fine-Tuning vs Transfer Learning
637
 
638
- Sube una imagen de papa (Chaucha o Chola) para compara el rendimiento de dos modelos entrenados.
639
- El sistema muestra resultados lado a lado, destacando con **borde dorado** 👑 el modelo con mayor confianza.
640
  """
641
  )
642
 
@@ -657,10 +657,10 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
657
  )
658
 
659
  boton_predecir = gr.Button(
660
- "🔍 Analizar con Ambos Modelos", variant="primary", size="lg"
661
  )
662
  boton_limpiar = gr.ClearButton(
663
- components=[imagen_entrada], value="🗑️ Limpiar", size="sm"
664
  )
665
 
666
  with gr.Column(scale=2):
@@ -674,7 +674,7 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
674
  gr.Markdown("### 🔥 Mapas de Atención (Grad-CAM)")
675
  with gr.Row():
676
  with gr.Column(scale=1):
677
- gr.Markdown("#### Modelo FT (Fine-Tuning)")
678
  imagen_gradcam_ft = gr.Image(label="Grad-CAM - Modelo FT", height=350)
679
 
680
  with gr.Column(scale=1):
@@ -692,8 +692,8 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
692
  ### Detalles Técnicos
693
 
694
  - **Modelos**:
695
- - **Modelo FT**: Fine-Tuning completo del ViT-Base/16
696
- - **Modelo TL**: Transfer Learning (solo cabeza final entrenada)
697
  - **Framework**: PyTorch + timm
698
  - **Dispositivo**: {DISPOSITIVO.upper()}
699
  - **Visualización**: EigenCAM (Grad-CAM para Transformers)
@@ -708,13 +708,13 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
708
 
709
  ### Criterio de "Ganador"
710
 
711
- El modelo con **mayor confianza** en su predicción se destaca con un **borde dorado** 👑.
712
  Esto no necesariamente significa que sea correcto, solo que está más seguro de su decisión.
713
 
714
  ### Sobre el Entrenamiento
715
 
716
- - **Modelo FT**: Todos los pesos ajustados con el dataset de papas
717
- - **Modelo TL**: Solo la capa de clasificación final entrenada
718
 
719
  Ambos modelos fueron entrenados con imágenes de tubérculos de papas Chaucha y Chola,
720
  capturando 3 tipos de defectos (brotes, podridas y cortadas) en condiciones de iluminación controlada.
 
632
  # Encabezado
633
  gr.Markdown(
634
  """
635
+ # 🥔 Clasificador de Calidad de Papas
636
+ ### Vision Transformer (ViT-Base/16) - Fine Tuning parcial y Transfer Learning
637
 
638
+ Sube una imagen de un tubérculo de papa (Chaucha o Chola) para compara el rendimiento de dos modelos entrenados.
639
+ El sistema muestra los resultados, destacando con **borde dorado** el modelo con mayor confianza.
640
  """
641
  )
642
 
 
657
  )
658
 
659
  boton_predecir = gr.Button(
660
+ "🔍 Clasificar", variant="primary", size="lg"
661
  )
662
  boton_limpiar = gr.ClearButton(
663
+ components=[imagen_entrada], value="🗑️ Limpiar", size="md"
664
  )
665
 
666
  with gr.Column(scale=2):
 
674
  gr.Markdown("### 🔥 Mapas de Atención (Grad-CAM)")
675
  with gr.Row():
676
  with gr.Column(scale=1):
677
+ gr.Markdown("#### Modelo FT (Fine Tuning parcial)")
678
  imagen_gradcam_ft = gr.Image(label="Grad-CAM - Modelo FT", height=350)
679
 
680
  with gr.Column(scale=1):
 
692
  ### Detalles Técnicos
693
 
694
  - **Modelos**:
695
+ - **Modelo FT**: Fine Tuning parcial (6 capas) del ViT-Base/16
696
+ - **Modelo TL**: Transfer Learning (solo cabeza clasificadora entrenada)
697
  - **Framework**: PyTorch + timm
698
  - **Dispositivo**: {DISPOSITIVO.upper()}
699
  - **Visualización**: EigenCAM (Grad-CAM para Transformers)
 
708
 
709
  ### Criterio de "Ganador"
710
 
711
+ El modelo con **mayor confianza** en su predicción se destaca con un **borde dorado**.
712
  Esto no necesariamente significa que sea correcto, solo que está más seguro de su decisión.
713
 
714
  ### Sobre el Entrenamiento
715
 
716
+ - **Modelo FT**: Los pesos de las 6 capas superiores han sido ajustados con el dataset de papas.
717
+ - **Modelo TL**: Solo la capa de clasificación final entrenada.
718
 
719
  Ambos modelos fueron entrenados con imágenes de tubérculos de papas Chaucha y Chola,
720
  capturando 3 tipos de defectos (brotes, podridas y cortadas) en condiciones de iluminación controlada.