update app.py

app.py

@@ -12,7 +12,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # 加载模型
 weights_path = "best_model.pth"
 model, _, _ = get_mobilenet_model(num_classes=16)
-model.load_state_dict(torch.load(weights_path, map_location=device, weights_only=False))
+checkpoint = torch.load(weights_path, map_location=device, weights_only=False)
+model.load_state_dict(checkpoint['state_dict'])  # 注意这里是 checkpoint['state_dict']
 model.to(device)
 model.eval()