feat: enable fine-tuned model loading from Hugging Face Hub

requirements.txt

@@ -1,5 +1,6 @@
 torch>=2.0.0
 transformers>=4.30.0
+peft>=0.4.0
 jsonschema>=4.0.0
 datasets>=2.0.0
 gradio>=5.0.0

test_constrained_model.py

@@ -37,14 +37,13 @@ def load_trained_model():
         
         # Try to load fine-tuned adapter from Hugging Face Hub
         try:
-            print("🔄 Attempting to load fine-tuned adapter...")
-            # from peft import PeftModel  # Uncomment when adapter is available
-            # model = PeftModel.from_pretrained(model, "jlov7/SmolLM3-Function-Calling-LoRA")
-            # model = model.merge_and_unload()
-            # print("✅ Fine-tuned model loaded successfully!")
-            print("🔧 Fine-tuned adapter not yet available - using base model with optimized prompting")
+            print("🔄 Attempting to load fine-tuned adapter from Hub...")
+            from peft import PeftModel
+            model = PeftModel.from_pretrained(model, "jlov7/SmolLM3-Function-Calling-LoRA")
+            model = model.merge_and_unload()
+            print("✅ Fine-tuned model loaded successfully from Hub!")
         except Exception as e:
-            print(f"⚠️ Could not load fine-tuned adapter: {e}")
+            print(f"⚠️ Could not load fine-tuned adapter from Hub: {e}")
             print("🔧 Using base model with optimized prompting")
         
         print("✅ Model loaded successfully")