Orphen2024 commited on
Commit
aee7b02
1 Parent(s): 07a27f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import json
6
 
7
  # Cargar la configuración
8
  #with open("reconocimiento_facial/config.json") as f:
9
- with open("CarPeAs/reconocimiento-facial/config.json") as f:
10
  config = json.load(f)
11
 
12
  # Definir las etiquetas de las clases
@@ -17,7 +17,7 @@ model = models.resnet50()
17
  model.fc = torch.nn.Linear(model.fc.in_features, len(class_labels)) # len(class_labels) = 7
18
 
19
  # Cargar los pesos del modelo
20
- model.load_state_dict(torch.load("CarPeAs/reconocimiento-facial/pytorch_model.bin"))
21
  model.eval()
22
 
23
  # Preprocesamiento de imágenes
 
6
 
7
  # Cargar la configuración
8
  #with open("reconocimiento_facial/config.json") as f:
9
+ with open("config.json") as f:
10
  config = json.load(f)
11
 
12
  # Definir las etiquetas de las clases
 
17
  model.fc = torch.nn.Linear(model.fc.in_features, len(class_labels)) # len(class_labels) = 7
18
 
19
  # Cargar los pesos del modelo
20
+ model.load_state_dict(torch.load("pytorch_model.bin"))
21
  model.eval()
22
 
23
  # Preprocesamiento de imágenes