from cloud_removal import DeepInpaintingStrategy
import torch
import numpy as np

cr = DeepInpaintingStrategy(model_path="cloud_removal_model/cloud_removal_unet_best.pth")
if cr.model is not None:
    expected = list(cr.model.parameters())[0].shape[1]
    print("Expected channels:", expected)
else:
    print("Failed to load model")