update rebuild

app.py

@@ -38,7 +38,7 @@ densenet, densenet_transforms = create_densenet121_model()
 # Load saved weights
 state_dict = torch.load("FL_global_model.pt", map_location=torch.device("cpu"))
 model_weights = state_dict["model"]
-densenet.load_state_dict(model_weights,strict=True)  # Set strict to True since we now expect it to match
+densenet.load_state_dict(model_weights,strict=False)  # Set strict to True since we now expect it to match
 
 def predict(img) -> Tuple[Dict, float]:
     """Transforms and performs a prediction on img and returns prediction and time taken."""