Update app.py (#1)

- Update app.py (5311e30a7c3813ccfb6ec2e310e948eb316b9d92)


Co-authored-by: Carlos Pérez <CarPeAs@users.noreply.huggingface.co>

app.py

@@ -17,7 +17,7 @@ model = models.resnet50()
 model.fc = torch.nn.Linear(model.fc.in_features, len(class_labels))  # len(class_labels) = 7
 
 # Cargar los pesos del modelo
-model.load_state_dict(torch.load("pytorch_model.bin"))
+model.load_state_dict(torch.load("pytorch_model.bin", map_location=torch.device('cpu')))
 model.eval()
 
 # Preprocesamiento de imágenes