Add application file

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Book Spine Detection
+emoji: 🏆
+colorFrom: gray
+colorTo: indigo
+sdk: streamlit
+sdk_version: 1.39.0
+app_file: app.py
+pinned: false
+short_description: Upload an image containing books, and this app will detect..
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,44 @@
+import streamlit as st
+from PIL import Image
+from ultralytics import YOLO
+
+# Load YOLOv8 model
+model = YOLO('best.pt')  # Ensure 'best.pt' is in the same directory
+
+# Professional message
+st.title("Book Detection with YOLOv8")
+st.markdown("""
+**Created by Siddharth Basale**
+
+Upload an image containing books, and this app will detect the number of books and generate an image with the bounding boxes.
+""")
+
+# Image uploader
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+if uploaded_file is not None:
+    # Load the image
+    image = Image.open(uploaded_file)
+    st.image(image, caption="Uploaded Image", use_column_width=True)
+
+    # Perform detection
+    results = model(image)
+
+    # Get the class ID for 'book'
+    book_class_id = None
+    for class_id, class_name in model.names.items():
+        if class_name == 'book':
+            book_class_id = class_id
+            break
+
+    # Count the number of books detected
+    if book_class_id is not None:
+        books_detected = len([r for r in results[0].boxes.data if int(r[-1]) == book_class_id])
+    else:
+        books_detected = 0  # No 'book' class detected in the model
+    
+    st.write(f"Number of books detected: {books_detected}")
+
+    # Display the image with bounding boxes
+    annotated_image = results[0].plot()  # Get image with bounding boxes
+    st.image(annotated_image, caption="Processed Image with Book Detection", use_column_width=True)

best.pt

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

requirements.txt

@@ -0,0 +1,3 @@
+ultralytics==8.3.27
+torch==2.5.1
+streamlit==1.39.0