Update README.md

README.md

@@ -116,7 +116,7 @@ from transformers import SiglipModel, SiglipProcessor
 from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
 
-class FaceRecognizer(nn.Module):
+class Model(nn.Module):
     def __init__(self):
         super().__init__()
         ckpt = "google/siglip-base-patch16-224"
@@ -128,7 +128,7 @@ class FaceRecognizer(nn.Module):
         clip_inputs = self.processor(images=images, return_tensors="pt").to(self.clip.device)
         return self.clip.get_image_features(**clip_inputs)
 
-model = FaceRecognizer()
+model = Model()
 
 weights_path = hf_hub_download(repo_id="AvitoTech/SigLIP-Base-for-animal-identification", filename="model.safetensors")
 state_dict = load_file(weights_path)