Update app.py
Browse files
app.py
CHANGED
|
@@ -31,12 +31,12 @@ CLASS_INDICES = {'Bean': 0,
|
|
| 31 |
'Tomato': 14}
|
| 32 |
|
| 33 |
@lru_cache(maxsize=1)
|
| 34 |
-
def load_model():
|
| 35 |
-
interpreter = lite.Interpreter(model_path=
|
| 36 |
print('loaded model')
|
| 37 |
return interpreter
|
| 38 |
|
| 39 |
-
interpreter = load_model(
|
| 40 |
|
| 41 |
|
| 42 |
def predict(test_image):
|
|
|
|
| 31 |
'Tomato': 14}
|
| 32 |
|
| 33 |
@lru_cache(maxsize=1)
|
| 34 |
+
def load_model(model_path):
|
| 35 |
+
interpreter = lite.Interpreter(model_path=model_path)
|
| 36 |
print('loaded model')
|
| 37 |
return interpreter
|
| 38 |
|
| 39 |
+
interpreter = load_model(model_path=os.path.join("models", "vegetable_classification_model_mnet.tflite"))
|
| 40 |
|
| 41 |
|
| 42 |
def predict(test_image):
|