test.py

import floret

# Load your pretrained model
model = floret.load_model("LID-40-3-2000000-1-4.bin")



print(model.get_labels())


import torch

# Try loading your Floret model with PyTorch


try:
    model = torch.load("LID-40-3-2000000-1-4.bin")
    print("Model successfully loaded as a PyTorch model.")
    print(model)
except Exception as e:
    print("Failed to load as a PyTorch model. Error:", e)