removes tflite error

adiazulay · adiazulay · commit a8b5d09ff2b1 · 2020-07-23T14:15:18.000-07:00
diff --git a/src/lobe/ImageModel.py b/src/lobe/ImageModel.py
@@ -17,7 +17,7 @@ def load_from_signature(signature: Signature) -> ImageModel:
     model_format = signature.format
     if model_format == "tf":
         from .backends import _backend_tf as backend
-    elif model_format == "tflite":
+    elif model_format == "tf_lite":
         from .backends import _backend_tflite as backend
     else:
         raise ValueError("Model is an unsupported format")
diff --git a/src/lobe/backends/_backend_tflite.py b/src/lobe/backends/_backend_tflite.py
@@ -22,7 +22,6 @@ def __init__(self, signature):
         self.__tflite_predict_fn = None
         self.__labels = signature.classes
 
-        raise ImportError("TFLite not yet supported")
 
     def __load(self):
         self.__tflite_predict_fn = tflite.Interpreter(
