Update main.py

main.py

@@ -261,7 +261,7 @@ def predict_readability(code):
             else:
                 outputs = model(**inputs)
         
-        amplified_logits = 4.0 * outputs.logits
+        amplified_logits = 2.0 * outputs.logits
         score = torch.sigmoid(amplified_logits).cpu().item()
         return round(max(0.0, min(1.0, score)), 4)
         
@@ -296,7 +296,7 @@ def predict_readability_batch(codes):
             else:
                 outputs = model(**inputs)
         
-        amplified_logits = 4.0 * outputs.logits
+        amplified_logits = 2.0 * outputs.logits
         scores = torch.sigmoid(amplified_logits).cpu().numpy()
 
         return [round(max(0.0, min(1.0, float(score))), 4) for score in scores.flatten()]