fixes: #19 : modified check for predict_fmlearn() check for model exi…

…stence

src/api.py

@@ -115,7 +115,7 @@ def predict_fmlearn():
 
     data = {}
 
-    if fml._new_recs == math.inf:
+    if not fml.is_model_trained():
         data['response'] = 'Model not trained!'
         return json.dumps(data)
 

src/fmlearn.py

@@ -43,6 +43,9 @@ def new_record_added(self):
         self._new_recs += 1
         return
 
+    def is_model_trained(self):
+        return self._model is not None
+
     def load_data(self):
         # force new model to be trained once data has been reloaded
         # retraining of model occurs only when the predict() function is being called