Update main.py

main.py

@@ -134,7 +134,7 @@ def predict_batch():
         if len(validated_codes) == 1:
             score = predict_vulnerability_with_chunking(validated_codes[0])
             cleanup_gpu_memory()
-            return jsonify({"results": [{"score": score}]})
+            return jsonify({"results": [{"score": 1.0 - score}]})
         
         batch_size = min(len(validated_codes), 16)
         results = []
@@ -187,7 +187,7 @@ def predict_batch():
                         batch_scores[local_idx] = 0.0
                 
                 for score in batch_scores:
-                    results.append({"score": score})
+                    results.append({"score": 1.0 - score})
                 
                 cleanup_gpu_memory()