change probability func

inference.py

@@ -165,7 +165,7 @@ def scaled_sigmoid(x, scale_factor=0.2, linear_property=0.3):
     # Combine sigmoid with linear component
     raw_prob = torch.sigmoid(scaled_x) * (1-linear_property) + linear_property * ((x + 25) / 50)
     # Clip to ensure bounds
-    return torch.clamp(raw_prob, min=0.011, max=0.989)
+    return torch.clamp(raw_prob, min=0.001, max=0.999)
 
 # Apply the scaled sigmoid