aliicemill commited on
Commit
0b5a4d3
1 Parent(s): 324d82b

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -32
app.py DELETED
@@ -1,32 +0,0 @@
1
- #!/usr/bin/env python
2
- # coding: utf-8
3
- #dosyayı py olarak kaydet ve komut satırını kullanarak streamlit run streamlit.py
4
- import streamlit as st
5
- from tensorflow.keras.models import load_model
6
- from PIL import Image
7
- import numpy as np
8
- import cv2
9
- model=load_model("animal_model.h5")
10
- def process_image(img):
11
- img=img.resize((224,224))
12
- img=np.array(img)
13
- img=img[:,:, :3] # Remove the alpha channel
14
- img=img/255.0
15
- img=np.expand_dims(img,axis=0)
16
- return img
17
- st.title('Animal Classification')
18
- st.write('Please choose an image so that the AI model can predict the type of animal.')
19
- file=st.file_uploader('Pick an image', type= ['jpg','jpeg','png'])
20
-
21
- with open("name of the animals.txt") as f:
22
- class_names = [x.strip() for x in f.readlines()]
23
-
24
- if file is not None:
25
- img=Image.open(file)
26
- st.image(img,caption='The image: ')
27
- image=process_image(img)
28
- prediction=model.predict(image)
29
- predicted_class=np.argmax(prediction)
30
- st.write('Probability Distribution')
31
- st.write(prediction)
32
- st.write("Prediction: ",class_names[predicted_class])