Update app.py

app.py

@@ -14,8 +14,17 @@ def analyze_model_parameters(model_path, hf_token=None, show_layer_details=False
         # Load model configuration first
         config = AutoConfig.from_pretrained(model_path, trust_remote_code=True, **token_kwargs)
         
-        # Load model on CPU
-        model = AutoModel.from_pretrained(model_path, device_map="cpu", trust_remote_code=True, **token_kwargs)
+        # Load model on CPU - FIXED: Use low_cpu_mem_usage=False to avoid meta tensors
+        model = AutoModel.from_pretrained(
+            model_path, 
+            torch_dtype=torch.float32,  # Explicitly set dtype
+            low_cpu_mem_usage=False,    # Disable meta tensors
+            trust_remote_code=True, 
+            **token_kwargs
+        )
+        
+        # Ensure model is on CPU
+        model = model.cpu()
         
         # Initialize counters
         total_params = 0