init

app.py

@@ -0,0 +1,35 @@
+from ultralytics import YOLO
+import streamlit as st
+from PIL import Image
+import openai
+from huggingface_hub import Repository
+
+repo = Repository("AmitPress/snake-detection")
+secret_key = repo.get_secret('openapi')
+openai.api_key(secret_key)
+model = YOLO('resource/twenty-eight-first.pt')
+
+classes = ['Ahaetulla nasuta', 'Ahaetulla prasina', 'Boiga cynodon', 'Boiga multomaculata', 'Bungarus caeruleus', 'Bungarus fasciatus', 'Chrysopelea ornata', 'Coelognathus radiatus', 'Daboia russelii', 'Dendrelaphis pictus', 'Dendrelaphis tristis', 'Echis carinatus', 'Fowlea piscator', 'Hydrophis platurus', 'Indotyphlops braminus', 'Laticauda colubrina', 'Lycodon aulicus', 'Malayopython reticulatus', 'Naja kaouthia', 'Naja naja', 'Oreocryptophis porphyraceus', 'Psammodynastes pulverulentus', 'Ptyas korros', 'Ptyas mucosa', 'Python bivittatus', 'Rhabdophis subminiatus', 'Trimeresurus albolabris', 'Xenochrophis piscator']
+col1, col2 = st.columns([2,1])
+
+col1.title('Snake Detection System')
+
+photo = col2.file_uploader('Upload an Image (JPEG)', type='jpg')
+if photo:
+    col2.success('Successfully Uploaded')
+    col1.image(photo)
+    image = Image.open(photo)
+    try:
+        bx = model(image)[0].boxes[0]
+        col1.write(f"Result: {(classes[int(bx.cls[0])])}")
+        txt = classes[int(bx.cls[0])]
+        genus, species = txt.split(' ')
+        venomousity = openai.Completion.create(engine="text-davinci-003", prompt=f"Is {genus} {species} is venomous? if venomous just say venomous otherwise say nonvenomous", max_tokens=50)
+        desc = openai.Completion.create(engine="text-davinci-003", prompt=f"Give a short description of {genus} {species}", max_tokens=500)
+        col1.write(venomousity)
+        col1.write(desc)
+    except:
+        col1.write(f"Error: There might be some network issue")
+else:
+    col1.write('Please Upload an Image of a Snake')
+    col1.write('Result will be shown as soon as the image is uploaded')

requirements.txt

@@ -0,0 +1,7 @@
+torch
+ultralytics
+pillow
+requests
+bs4
+openai
+huggingface_hub

resource/twenty-eight-first.pt

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