reformat model impl

google_perch_lite/model.py

@@ -1,17 +1,19 @@
 from iSparrow.sparrow_model_base import ModelBase
-try: 
+
+try:
     import tflite_runtime.interpreter as tflite
 except ImportError:
     import tensorflow.lite as tflite
 
-from iSparrow import utils 
-from iSparrow import ModelBase 
+from iSparrow import utils
+from iSparrow import ModelBase
 
 import numpy as np
 from pathlib import Path
 from scipy.special import softmax
 
-class Model(ModelBase): 
+
+class Model(ModelBase):
     """
     Model Implementation of a iSparrow model that uses the google perch tflite model.
 
@@ -19,7 +21,7 @@ class Model(ModelBase):
         ModelBase (iSparrow.ModelBase): Model base class that provides the interface through which to interact with iSparrow.
     """
 
-    def __init__(self, model_path: str, num_threads: int = 1, **kwargs): 
+    def __init__(self, model_path: str, num_threads: int = 1, **kwargs):
         """
         __init__ Create a new model instance that uses the google perch tflite converted model.
 
@@ -48,7 +50,6 @@ class Model(ModelBase):
 
         self.output_layer_index = output_details[1]["index"]
 
-
     def predict(self, sample: np.array) -> np.array:
         """
         predict Make inference about the bird species for the preprocessed data passed to this function as arguments.
@@ -75,7 +76,6 @@ class Model(ModelBase):
 
         return confidence
 
-
     @classmethod
     def from_cfg(cls, sparrow_folder: str, cfg: dict):
         """