Metinhsimi commited on
Commit
09ba665
Β·
verified Β·
1 Parent(s): 40fd5b5

Upload 3 files

Browse files
Files changed (3) hide show
  1. app.py +63 -0
  2. model_butter.h5 +3 -0
  3. requirements.txt +2 -0
app.py ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from tensorflow.keras.preprocessing import image
3
+ import numpy as np
4
+ from PIL import Image
5
+ import tensorflow as tf
6
+
7
+ # Load the model
8
+ @st.cache_resource
9
+ def load_model():
10
+ model = tf.keras.models.load_model('model_butter.h5') # Path to your model
11
+ return model
12
+
13
+ model = load_model()
14
+
15
+ # Class labels
16
+ class_labels = [
17
+ 'SOUTHERN DOGFACE', 'ADONIS', 'BROWN SIPROETA', 'MONARCH', 'GREEN CELLED CATTLEHEART',
18
+ 'CAIRNS BIRDWING', 'EASTERN DAPPLE WHITE', 'RED POSTMAN', 'MANGROVE SKIPPER',
19
+ 'BLACK HAIRSTREAK', 'CABBAGE WHITE', 'RED ADMIRAL', 'PAINTED LADY', 'PAPER KITE',
20
+ 'SOOTYWING', 'PINE WHITE', 'PEACOCK', 'CHECQUERED SKIPPER', 'JULIA',
21
+ 'COMMON WOOD-NYMPH', 'BLUE MORPHO', 'CLOUDED SULPHUR', 'STRAITED QUEEN',
22
+ 'ORANGE OAKLEAF', 'PURPLISH COPPER', 'ATALA', 'IPHICLUS SISTER', 'DANAID EGGFLY',
23
+ 'LARGE MARBLE', 'PIPEVINE SWALLOW', 'BLUE SPOTTED CROW', 'RED CRACKER',
24
+ 'QUESTION MARK', 'CRIMSON PATCH', 'BANDED PEACOCK', 'SCARCE SWALLOW', 'COPPER TAIL',
25
+ 'GREAT JAY', 'INDRA SWALLOW', 'VICEROY', 'MALACHITE', 'APPOLLO',
26
+ 'TWO BARRED FLASHER', 'MOURNING CLOAK', 'TROPICAL LEAFWING', 'POPINJAY',
27
+ 'ORANGE TIP', 'GOLD BANDED', 'BECKERS WHITE', 'RED SPOTTED PURPLE',
28
+ 'MILBERTS TORTOISESHELL', 'SILVER SPOT SKIPPER', 'AMERICAN SNOOT', 'AN 88',
29
+ 'ULYSES', 'COMMON BANDED AWL', 'CRECENT', 'METALMARK', 'SLEEPY ORANGE',
30
+ 'PURPLE HAIRSTREAK', 'ELBOWED PIERROT', 'GREAT EGGFLY', 'ORCHARD SWALLOW',
31
+ 'ZEBRA LONG WING', 'WOOD SATYR', 'MESTRA', 'EASTERN PINE ELFIN',
32
+ 'EASTERN COMA', 'YELLOW SWALLOW TAIL', 'CLEOPATRA', 'GREY HAIRSTREAK',
33
+ 'BANDED ORANGE HELICONIAN', 'AFRICAN GIANT SWALLOWTAIL', 'CHESTNUT',
34
+ 'CLODIUS PARNASSIAN'
35
+ ]
36
+
37
+ # Title
38
+ st.title("πŸ¦‹ Butterfly Classifier")
39
+
40
+ # Image Upload
41
+ st.header("Upload an Image")
42
+ uploaded_file = st.file_uploader("Please upload a butterfly image...", type=["jpg", "jpeg", "png"])
43
+
44
+ if uploaded_file is not None:
45
+ # Open and display the image
46
+ img = Image.open(uploaded_file)
47
+ st.image(img, caption='Uploaded Image', use_column_width=True)
48
+ st.write("πŸ” **Analyzing the image...**")
49
+
50
+ # Preprocess the image
51
+ img = img.resize((128, 128)) # Resize according to how your model was trained
52
+ img_array = image.img_to_array(img)
53
+ img_array = np.expand_dims(img_array, axis=0)
54
+ img_array /= 255.0
55
+
56
+ # Predict
57
+ prediction = model.predict(img_array)
58
+ predicted_class = np.argmax(prediction, axis=1)[0]
59
+
60
+ # Display the result
61
+ st.write(f"πŸ¦‹ **It's a {class_labels[predicted_class]}!**")
62
+ else:
63
+ st.write("πŸ‘ˆ Upload an image to get started!")
model_butter.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d6e9c5d2331719a9a2c6d89b1657de8be1101c9bb50c49a3d7575f05989e4a3
3
+ size 45063260
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ tensorflow
2
+ streamlit