Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 def count_parameters(model_path):
     try:
         # Load model on CPU
-        model = AutoModel.from_pretrained(model_path, device_map="cpu")
+        model = AutoModel.from_pretrained(model_path, device_map="cpu", trust_remote_code=True)
         
         # Count trainable parameters (accounting for weight tying)
         unique_params = {}