handex commited on
Commit
f4be932
·
verified ·
1 Parent(s): b9a1780

Delete src/fruit_1.py

Browse files
Files changed (1) hide show
  1. src/fruit_1.py +0 -32
src/fruit_1.py DELETED
@@ -1,32 +0,0 @@
1
- import streamlit as st
2
- from tensorflow.keras.models import load_model
3
- from PIL import Image
4
- import numpy as np
5
- import cv2
6
-
7
- model=load_model('fruit_vegetable.h5')
8
-
9
- def process_image(img):
10
- img=img.resize((240,240))
11
- img=np.array(img)
12
- img=img/255.0
13
- img=np.expand_dims(img,axis=0)
14
- return img
15
-
16
- st.title("Meyve/Sebze sınıflandırma")
17
- st.write("Resmini yükle ve model meyve/sebze ismini tahmin etsin")
18
-
19
- file=st.file_uploader('Bir resim sec',type=['jpg','jpeg','png'])
20
-
21
- if file is not None:
22
- img=Image.open(file)
23
- st.image(img,caption='Meyve/Sebze')
24
- image= process_image(img)
25
- prediction=model.predict(image)
26
- predicted_class=np.argmax(prediction)
27
-
28
- class_names=['capsicum','sweetcorn','orange','tomato','turnip','ginger','raddish','pomegranate','pineapple',
29
- 'jalepeno','apple','carrot','lettuce','bell pepper','eggplant','beetroot','kiwi','pear',
30
- 'cabbage','cauliflower','paprika','lemon','sweetpotato','grapes','cucumber','corn','banana',
31
- 'garlic','chilli pepper','watermelon','mango','peas','onion','potato','spinach','soy beans']
32
- st.write(class_names[predicted_class])