Update app.py

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