Upload test_model.py with huggingface_hub

test_model.py

@@ -1,8 +1,14 @@
 #!/usr/bin/env python3
+# Test script for the DeepSeek D&D LoRA model
+# This script uses the Hugging Face Hub to track usage metrics
+
+from huggingface_hub import snapshot_download
 import torch
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
+print("Loading model and tokenizer...")
+
 # Load the base model and tokenizer
 base_model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B"
 model = AutoModelForCausalLM.from_pretrained(
@@ -12,8 +18,8 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 tokenizer = AutoTokenizer.from_pretrained(base_model_id)
 
-# Load the LoRA adapter
-adapter_model_id = "chad/deepseek-dnd-lora"
+# Load the LoRA adapter - this will be tracked by HF Hub
+adapter_model_id = "chendren/deepseek-dnd-lora"
 model = PeftModel.from_pretrained(model, adapter_model_id)
 
 # Test prompts