oguztoy commited on
Commit
d190575
·
verified ·
1 Parent(s): 68cf33a

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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 Classifier")
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