handex commited on
Commit
c80ffc3
·
verified ·
1 Parent(s): 4e8816b

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +31 -39
src/streamlit_app.py CHANGED
@@ -1,40 +1,32 @@
1
- import altair as alt
2
- import numpy as np
3
- import pandas as pd
4
  import streamlit as st
5
-
6
- """
7
- # Welcome to Streamlit!
8
-
9
- Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
10
- If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
11
- forums](https://discuss.streamlit.io).
12
-
13
- In the meantime, below is an example of what you can do with just a few lines of code:
14
- """
15
-
16
- num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
17
- num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
18
-
19
- indices = np.linspace(0, 1, num_points)
20
- theta = 2 * np.pi * num_turns * indices
21
- radius = indices
22
-
23
- x = radius * np.cos(theta)
24
- y = radius * np.sin(theta)
25
-
26
- df = pd.DataFrame({
27
- "x": x,
28
- "y": y,
29
- "idx": indices,
30
- "rand": np.random.randn(num_points),
31
- })
32
-
33
- st.altair_chart(alt.Chart(df, height=700, width=700)
34
- .mark_point(filled=True)
35
- .encode(
36
- x=alt.X("x", axis=None),
37
- y=alt.Y("y", axis=None),
38
- color=alt.Color("idx", legend=None, scale=alt.Scale()),
39
- size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
40
- ))
 
 
 
 
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])