Fix clinical analysis to use 'out' key from finetuned model (actual working output)

clinical_analysis.py

@@ -35,6 +35,23 @@ def analyze_ecg_features(model_output: Dict[str, Any]) -> Dict[str, Any]:
             clinical_result = extract_clinical_from_probabilities(probs)
             return clinical_result
         
+        # NEW: Check for 'out' key (actual finetuned model output)
+        elif 'out' in model_output:
+            print("✅ Found 'out' key - using finetuned model output")
+            # FINETUNED MODEL - Extract real clinical predictions
+            logits = model_output['out']
+            if isinstance(logits, torch.Tensor):
+                # Remove batch dimension if present
+                if logits.dim() == 2:  # [batch, num_labels]
+                    logits = logits.squeeze(0)  # Remove batch dimension
+                probs = torch.sigmoid(logits).detach().cpu().numpy().ravel()
+            else:
+                probs = 1 / (1 + np.exp(-np.array(logits).ravel()))
+            
+            # Extract clinical parameters from probabilities
+            clinical_result = extract_clinical_from_probabilities(probs)
+            return clinical_result
+        
         # NEW: Check if the model output IS the logits tensor directly (classifier model)
         elif isinstance(model_output, torch.Tensor):
             print("✅ Found direct logits tensor - using classifier model output")