Experiment

Model_Handler.py

@@ -9,8 +9,8 @@ class Model_handler :
     print("Model loaded and is ready to use !!!")
 
   def predict(self, state, color) :
-    prediction = self.model(self.convert_state_to_input(state, color))
-    return self.convert_output_to_probs(state, color, prediction[1][0])
+    prediction = self.model.predict(self.convert_state_to_input(state, color))
+    return self.convert_output_to_probs(state, color, prediction[0])
 
   def convert_state_to_input(self, state, color) :
     if type(state) == str :