Update app.py

app.py

@@ -11,7 +11,6 @@ keras.utils.get_custom_objects()['recall'] = utils.recall
 keras.utils.get_custom_objects()['precision'] = utils.precision
 keras.utils.get_custom_objects()['f1'] = utils.f1
 
-
 def teachable_machine_classification(img=None, model=None):
     """Performs inference on image uploaded"""
 
@@ -34,7 +33,7 @@ def teachable_machine_classification(img=None, model=None):
     # run the inference
     prediction = model.predict(data)
 
-    print("Prediction", prediction)
+    logging.info("Prediction", prediction)
     return prediction[0][
         1
     ]  # np.argmax(prediction) # return position of the highest probability
@@ -54,7 +53,7 @@ def load_model(weights_file=None):
         model = keras.models.load_model(
             weights_file, custom_objects=dependencies, compile=False
         )
-
+        logging.info("Model successfully loaded.")
         return model
     except Exception as e:
         logging.error("ERROR: ", e)