from basicsr.models.archs.histoformer_arch import Histoformer
# from huggingface_hub import PyTorchModelHubMixin
from huggingface_hub import hf_hub_download
# define model
model = Histoformer()

# equip with weights
filepath = hf_hub_download(repo_id="sunsean/Histoformer", filename="Allweather/pretrained_models/net_g_real.pth")
model.load_state_dict(filepath, map_location="gpu")

# push to hub
model.push_to_hub("sunsean/Histoformer-real")

# reload
model = ImageCleanModel.from_pretrained("sunsean/Histoformer-real")