using different model

app.py

@@ -18,10 +18,11 @@ model = LlamaForCausalLM(
 )
 device = "cpu"
 
-checkpoint_path = "model_bin.pth"
+model_id = "satyanayak/custom-smallmv2135"
+checkpoint_path = hf_hub_download(repo_id=model_id, filename="model-dict-step-5500.pt")
+
 checkpoint = torch.load(checkpoint_path, map_location=device)
-#model.load_state_dict(checkpoint['model_state_dict'])
-model.load_state_dict(torch.load(checkpoint_path, map_location=device))
+model.load_state_dict(checkpoint['model_state_dict'])
 model.to(device)
 model.eval()