Update app.py

app.py

@@ -3,14 +3,14 @@ import gradio as gr
 from pathlib import Path
 import zipfile
 
-# Extraer dataset.zip al iniciar la app
+# Extraer dataset.zip al inicio
 with zipfile.ZipFile("dataset.zip", 'r') as zip_ref:
-    zip_ref.extractall(".")  # crea la carpeta dataset/
+    zip_ref.extractall(".")  # crea la carpeta dataset/ directamente
 
 # Ruta relativa al dataset extraído
 path = Path("dataset")
 
-# Crear DataLoaders desde las carpetas de clases
+# Crear DataLoaders desde las subcarpetas de clases
 dls = ImageDataLoaders.from_folder(
     path,
     valid_pct=0.2,
@@ -20,7 +20,7 @@ dls = ImageDataLoaders.from_folder(
 
 # Cargar el modelo .pth
 learn = vision_learner(dls, resnet34)
-learn.load("model_lab")  # si lo guardaste en models/, usar "models/model_lab"
+learn.load("model_lab")  # si lo pusiste en models/, usar "models/model_lab"
 
 # Tomar automáticamente las clases desde los DataLoaders
 labels = learn.dls.vocab
@@ -40,3 +40,4 @@ demo = gr.Interface(
 )
 
 demo.launch()
+