Views
No views yet
1import torch
2from huggingface_hub import hf_hub_download
3
4# Download model
5model_path = hf_hub_download(
6 repo_id="kristiangnordby/gemma_cyber",
7 filename="model.pt"
8)
9
10# Load model
11checkpoint = torch.load(model_path, map_location='cpu')
12
13# For inference, you'll need the model class definition
14# See model_architecture.py in this repo1@misc{cybersecurity_classifier,
2 author = {Kristian Nordby},
3 title = {Cybersecurity Binary Classifier},
4 year = {2025},
5 publisher = {HuggingFace},
6 howpublished = {\url{https://huggingface.co/kristiangnordby/gemma_cyber}}
7}