JuanSv commited on
Commit
d9057eb
·
verified ·
1 Parent(s): bf8d62d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -3,9 +3,9 @@ import gradio as gr
3
  from pathlib import Path
4
  import zipfile
5
 
6
- # Extraer dataset.zip al inicio
7
  with zipfile.ZipFile("dataset.zip", 'r') as zip_ref:
8
- zip_ref.extractall(".") # crea la carpeta dataset
9
 
10
  # Ruta relativa al dataset extraído
11
  path = Path("dataset")
@@ -20,7 +20,7 @@ dls = ImageDataLoaders.from_folder(
20
 
21
  # Cargar el modelo .pth
22
  learn = vision_learner(dls, resnet34)
23
- learn.load("model_lab") # si está en models/, usar "models/model_lab"
24
 
25
  # Tomar automáticamente las clases desde los DataLoaders
26
  labels = learn.dls.vocab
@@ -40,5 +40,3 @@ demo = gr.Interface(
40
  )
41
 
42
  demo.launch()
43
-
44
-
 
3
  from pathlib import Path
4
  import zipfile
5
 
6
+ # Extraer dataset.zip al iniciar la app
7
  with zipfile.ZipFile("dataset.zip", 'r') as zip_ref:
8
+ zip_ref.extractall(".") # crea la carpeta dataset/
9
 
10
  # Ruta relativa al dataset extraído
11
  path = Path("dataset")
 
20
 
21
  # Cargar el modelo .pth
22
  learn = vision_learner(dls, resnet34)
23
+ learn.load("model_lab") # si lo guardaste en models/, usar "models/model_lab"
24
 
25
  # Tomar automáticamente las clases desde los DataLoaders
26
  labels = learn.dls.vocab
 
40
  )
41
 
42
  demo.launch()