Missing import for inference (#5)

- Missing import for inference (4835c877cf34613645833d264696024abd456a82)


Co-authored-by: Davide Cifarelli <davide221@users.noreply.huggingface.co>

README.md

@@ -47,6 +47,7 @@ import os
 import torch
 from torch import nn
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import snapshot_download
 
 ## Define the reward model function class