ksvmuralidhar commited on
Commit
d64c5a6
1 Parent(s): ed885a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=os.path.join("models", "vegetable_classification_model_mnet.tflite"))
36
  print('loaded model')
37
  return interpreter
38
 
39
- interpreter = load_model(True)
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):