Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
|
@@ -5,7 +5,7 @@ from tensorflow.keras.preprocessing import image
|
|
| 5 |
from tensorflow.keras.applications.efficientnet import preprocess_input
|
| 6 |
from PIL import Image
|
| 7 |
|
| 8 |
-
model = load_model("best_efficientnet_model.keras")
|
| 9 |
|
| 10 |
class_names = [
|
| 11 |
'apple', 'banana', 'beetroot', 'bell pepper', 'cabbage', 'capsicum', 'carrot',
|
|
@@ -15,7 +15,7 @@ class_names = [
|
|
| 15 |
'soy beans', 'spinach', 'sweetcorn', 'sweetpotato', 'tomato', 'turnip', 'watermelon'
|
| 16 |
]
|
| 17 |
|
| 18 |
-
st.title("Fruit and Vegetable
|
| 19 |
|
| 20 |
st.markdown("Upload an image of a fruit or vegetable. The model will predict its class.")
|
| 21 |
|
|
|
|
| 5 |
from tensorflow.keras.applications.efficientnet import preprocess_input
|
| 6 |
from PIL import Image
|
| 7 |
|
| 8 |
+
model = load_model("src/best_efficientnet_model.keras")
|
| 9 |
|
| 10 |
class_names = [
|
| 11 |
'apple', 'banana', 'beetroot', 'bell pepper', 'cabbage', 'capsicum', 'carrot',
|
|
|
|
| 15 |
'soy beans', 'spinach', 'sweetcorn', 'sweetpotato', 'tomato', 'turnip', 'watermelon'
|
| 16 |
]
|
| 17 |
|
| 18 |
+
st.title("Fruit and Vegetable Recognition")
|
| 19 |
|
| 20 |
st.markdown("Upload an image of a fruit or vegetable. The model will predict its class.")
|
| 21 |
|