From 6be1268feccfd21f4482a5957f72221347588893 Mon Sep 17 00:00:00 2001
From: Ludovic Moncla <moncla.ludovic@gmail.com>
Date: Fri, 25 Nov 2022 14:39:08 +0100
Subject: [PATCH] Update Predict_LGE.py

---
 scripts/Predict_LGE.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/Predict_LGE.py b/scripts/Predict_LGE.py
index fabc4cf..7d474e1 100644
--- a/scripts/Predict_LGE.py
+++ b/scripts/Predict_LGE.py
@@ -186,7 +186,7 @@ if __name__ == '__main__':
   pred = predict(model, data_loader, device)
 
   encoder_filename = "models/label_encoder.pkl"
-  with open(path+encoder_filename, 'rb') as file:
+  with open(path + encoder_filename, 'rb') as file:
         encoder = pickle.load(file)
   
   p2 = list(encoder.inverse_transform(pred))
-- 
GitLab