diff --git a/utils/lgb_predict.py b/utils/lgb_predict.py
index 1546b0254586f23d13831b3ccf672ddc608f027f..010cab89c233691611b40dbcb9f4584dd3ae807d 100755
--- a/utils/lgb_predict.py
+++ b/utils/lgb_predict.py
@@ -32,7 +32,7 @@ LGB_CONFIG = {
 MAP = ["irrelevant", "complement", "substitute", "exact"]
 LOCALE_MAP = {'jp':0, 'es':1, 'us':2}
 COL_NAME = "esci_label"
-WEIGHT_MAP = {0:0.4, 1:0.4, 2:0.4, 3:0.4, 4:0.2, 5:0.2}
+WEIGHT_MAP = {0:0.4, 1:0.4, 2:0.4, 3:0.4, 4:0.25, 5:0.2}
 
 def lgb_predict(df, locale):
     df = df.reset_index(drop=True)
diff --git a/utils/run.py b/utils/run.py
index 94a5ab65f0b581f2d546f3677e77140e295c4719..2bac4919649e4b71ed35aa7b9fd20268380b91b1 100755
--- a/utils/run.py
+++ b/utils/run.py
@@ -57,7 +57,8 @@ CONFIG = {
                 "product_brand": 42452,
                 "product_color_name": 44287,
             },
-            "model": ["/models/us-distilbart-two-fold-0-fp16.onnx" ,"/models/us-distilbart-two-fold-1-fp16.onnx"],
+            "model": ["/models/us-distilbart-two-fold-0-fp16.onnx"],
+            #"model": ["/models/us-distilbart-two-fold-0-fp16.onnx" ,"/models/us-distilbart-two-fold-1-fp16.onnx"],
             "product": "/models/distilbart.h5",
             "key": "us",
             "type": "distilbart",