Upload 7 files

Dockerfile

@@ -1,21 +1,21 @@
-FROM python:3.9-slim
-
-WORKDIR /app
-
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
-    && rm -rf /var/lib/apt/lists/*
-
-COPY requirements.txt ./
-COPY src/ ./src/
-
-RUN pip3 install -r requirements.txt
-
-EXPOSE 8501
-
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
-
+FROM python:3.9-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    software-properties-common \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt ./
+COPY src/ ./src/
+
+RUN pip3 install -r requirements.txt
+
+EXPOSE 8501
+
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+
 ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]

README.md

@@ -1,19 +1,19 @@
----
-title: Libkamaja Id
-emoji: 🚀
-colorFrom: red
-colorTo: red
-sdk: docker
-app_port: 8501
-tags:
-- streamlit
-pinned: false
-short_description: Tööriist liblikat täpseks tuvastamiseks liblikamaja
----
-
-# Welcome to Streamlit!
-
-Edit `/src/streamlit_app.py` to customize this app to your heart's desire. :heart:
-
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+---
+title: Libkamaja Id
+emoji: 🚀
+colorFrom: red
+colorTo: red
+sdk: docker
+app_port: 8501
+tags:
+- streamlit
+pinned: false
+short_description: Tööriist liblikat täpseks tuvastamiseks liblikamaja
+---
+
+# Welcome to Streamlit!
+
+Edit `/src/streamlit_app.py` to customize this app to your heart's desire. :heart:
+
+If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
+forums](https://discuss.streamlit.io).

app.py

@@ -0,0 +1,55 @@
+
+import streamlit as st
+from PIL import Image
+import torch
+from torchvision import models, transforms
+import json
+
+# Load class names
+with open("class_names.txt", "r") as f:
+    class_names = [line.strip() for line in f.readlines()]
+
+# Load butterfly info
+try:
+    with open("butterfly_info.json", "r") as f:
+        butterfly_info = json.load(f)
+except:
+    butterfly_info = {}
+
+@st.cache_resource
+def load_model():
+    model = models.resnet18(pretrained=False)
+    model.fc = torch.nn.Linear(model.fc.in_features, len(class_names))
+    model.load_state_dict(torch.load("/content/drive/MyDrive/butterfly_classifier.pth", map_location="cpu"))
+    model.eval()
+    return model
+
+model = load_model()
+
+transform = transforms.Compose([
+    transforms.Resize((224, 224)),
+    transforms.ToTensor(),
+])
+
+st.title("🦋 Butterfly Identifier")
+st.write("Upload a butterfly image and I’ll tell you what species it is!")
+
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+if uploaded_file:
+    image = Image.open(uploaded_file).convert("RGB")
+    st.image(image, caption="Uploaded Image", use_column_width=True)
+
+    # Preprocess
+    input_tensor = transform(image).unsqueeze(0)
+
+    # Predict
+    with torch.no_grad():
+        output = model(input_tensor)
+        _, pred = torch.max(output, 1)
+        predicted_class = class_names[pred.item()]
+
+    st.success(f"**Prediction: {predicted_class}**")
+
+    if predicted_class in butterfly_info:
+        st.info(butterfly_info[predicted_class]["description"])

butterfly_classifier.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b94b31c23255ed57a5181d92034cede75677504fbf3e5604d6a424b70d923c82
+size 44887396

butterfly_info.json

@@ -0,0 +1,152 @@
+{
+  "adonis": {
+    "description": "adonis is a butterfly species. (Add more info here.)"
+  },
+  "american snoot": {
+    "description": "american snoot is a butterfly species. (Add more info here.)"
+  },
+  "an 88": {
+    "description": "an 88 is a butterfly species. (Add more info here.)"
+  },
+  "banded peacock": {
+    "description": "banded peacock is a butterfly species. (Add more info here.)"
+  },
+  "beckers white": {
+    "description": "beckers white is a butterfly species. (Add more info here.)"
+  },
+  "black hairstreak": {
+    "description": "black hairstreak is a butterfly species. (Add more info here.)"
+  },
+  "cabbage white": {
+    "description": "cabbage white is a butterfly species. (Add more info here.)"
+  },
+  "chestnut": {
+    "description": "chestnut is a butterfly species. (Add more info here.)"
+  },
+  "clodius parnassian": {
+    "description": "clodius parnassian is a butterfly species. (Add more info here.)"
+  },
+  "clouded sulphur": {
+    "description": "clouded sulphur is a butterfly species. (Add more info here.)"
+  },
+  "copper tail": {
+    "description": "copper tail is a butterfly species. (Add more info here.)"
+  },
+  "crecent": {
+    "description": "crecent is a butterfly species. (Add more info here.)"
+  },
+  "crimson patch": {
+    "description": "crimson patch is a butterfly species. (Add more info here.)"
+  },
+  "eastern coma": {
+    "description": "eastern coma is a butterfly species. (Add more info here.)"
+  },
+  "gold banded": {
+    "description": "gold banded is a butterfly species. (Add more info here.)"
+  },
+  "great eggfly": {
+    "description": "great eggfly is a butterfly species. (Add more info here.)"
+  },
+  "grey hairstreak": {
+    "description": "grey hairstreak is a butterfly species. (Add more info here.)"
+  },
+  "indra swallow": {
+    "description": "indra swallow is a butterfly species. (Add more info here.)"
+  },
+  "julia": {
+    "description": "julia is a butterfly species. (Add more info here.)"
+  },
+  "large marble": {
+    "description": "large marble is a butterfly species. (Add more info here.)"
+  },
+  "malachite": {
+    "description": "malachite is a butterfly species. (Add more info here.)"
+  },
+  "mangrove skipper": {
+    "description": "mangrove skipper is a butterfly species. (Add more info here.)"
+  },
+  "metalmark": {
+    "description": "metalmark is a butterfly species. (Add more info here.)"
+  },
+  "monarch": {
+    "description": "monarch is a butterfly species. (Add more info here.)"
+  },
+  "morning cloak": {
+    "description": "morning cloak is a butterfly species. (Add more info here.)"
+  },
+  "orange oakleaf": {
+    "description": "orange oakleaf is a butterfly species. (Add more info here.)"
+  },
+  "orange tip": {
+    "description": "orange tip is a butterfly species. (Add more info here.)"
+  },
+  "orchard swallow": {
+    "description": "orchard swallow is a butterfly species. (Add more info here.)"
+  },
+  "painted lady": {
+    "description": "painted lady is a butterfly species. (Add more info here.)"
+  },
+  "paper kite": {
+    "description": "paper kite is a butterfly species. (Add more info here.)"
+  },
+  "peacock": {
+    "description": "peacock is a butterfly species. (Add more info here.)"
+  },
+  "pine white": {
+    "description": "pine white is a butterfly species. (Add more info here.)"
+  },
+  "pipevine swallow": {
+    "description": "pipevine swallow is a butterfly species. (Add more info here.)"
+  },
+  "purple hairstreak": {
+    "description": "purple hairstreak is a butterfly species. (Add more info here.)"
+  },
+  "question mark": {
+    "description": "question mark is a butterfly species. (Add more info here.)"
+  },
+  "red admiral": {
+    "description": "red admiral is a butterfly species. (Add more info here.)"
+  },
+  "red spotted purple": {
+    "description": "red spotted purple is a butterfly species. (Add more info here.)"
+  },
+  "scarce swallow": {
+    "description": "scarce swallow is a butterfly species. (Add more info here.)"
+  },
+  "silver spot skipper": {
+    "description": "silver spot skipper is a butterfly species. (Add more info here.)"
+  },
+  "sixspot burnet": {
+    "description": "sixspot burnet is a butterfly species. (Add more info here.)"
+  },
+  "skipper": {
+    "description": "skipper is a butterfly species. (Add more info here.)"
+  },
+  "sootywing": {
+    "description": "sootywing is a butterfly species. (Add more info here.)"
+  },
+  "southern dogface": {
+    "description": "southern dogface is a butterfly species. (Add more info here.)"
+  },
+  "straited queen": {
+    "description": "straited queen is a butterfly species. (Add more info here.)"
+  },
+  "two barred flasher": {
+    "description": "two barred flasher is a butterfly species. (Add more info here.)"
+  },
+  "ulyses": {
+    "description": "ulyses is a butterfly species. (Add more info here.)"
+  },
+  "viceroy": {
+    "description": "viceroy is a butterfly species. (Add more info here.)"
+  },
+  "wood satyr": {
+    "description": "wood satyr is a butterfly species. (Add more info here.)"
+  },
+  "yellow swallow tail": {
+    "description": "yellow swallow tail is a butterfly species. (Add more info here.)"
+  },
+  "zebra long wing": {
+    "description": "zebra long wing is a butterfly species. (Add more info here.)"
+  }
+}

class_names.txt

@@ -0,0 +1,50 @@
+adonis
+american snoot
+an 88
+banded peacock
+beckers white
+black hairstreak
+cabbage white
+chestnut
+clodius parnassian
+clouded sulphur
+copper tail
+crecent
+crimson patch
+eastern coma
+gold banded
+great eggfly
+grey hairstreak
+indra swallow
+julia
+large marble
+malachite
+mangrove skipper
+metalmark
+monarch
+morning cloak
+orange oakleaf
+orange tip
+orchard swallow
+painted lady
+paper kite
+peacock
+pine white
+pipevine swallow
+purple hairstreak
+question mark
+red admiral
+red spotted purple
+scarce swallow
+silver spot skipper
+sixspot burnet
+skipper
+sootywing
+southern dogface
+straited queen
+two barred flasher
+ulyses
+viceroy
+wood satyr
+yellow swallow tail
+zebra long wing

requirements.txt

@@ -1,3 +1,6 @@
-altair
-pandas
-streamlit
+altair
+pandas
+streamlit
+torch
+torchvision
+pillow