Spaces:
Runtime error
Runtime error
Orphen2024
commited on
Commit
•
aee7b02
1
Parent(s):
07a27f0
Update app.py
Browse files
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("
|
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("
|
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
|