Spaces:
Sleeping
Sleeping
ksvmuralidhar
commited on
Commit
•
8b93c35
1
Parent(s):
bd7da7f
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,12 @@ CLASS_INDICES = {'Bean': 0,
|
|
29 |
'Radish': 13,
|
30 |
'Tomato': 14}
|
31 |
|
32 |
-
|
|
|
|
|
|
|
|
|
|
|
33 |
|
34 |
|
35 |
def predict(test_image):
|
|
|
29 |
'Radish': 13,
|
30 |
'Tomato': 14}
|
31 |
|
32 |
+
@st.cache_resource
|
33 |
+
def load_model():
|
34 |
+
interpreter = lite.Interpreter(model_path=os.path.join("models", "vegetable_classification_model_mnet.tflite"))
|
35 |
+
return interpreter
|
36 |
+
|
37 |
+
interpreter = load_model()
|
38 |
|
39 |
|
40 |
def predict(test_image):
|