Add: app v.01

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Ilia Munaev
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,14 +1 @@
----
-title: Chess Ai Classifier
-emoji: 😻
-colorFrom: red
-colorTo: indigo
-sdk: gradio
-sdk_version: 5.44.1
-app_file: app.py
-pinned: false
-license: mit
-short_description: A deep learning project that recognizes chess pieces.
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# chess-ai-classifier

app.py

@@ -0,0 +1,28 @@
+from fastai.vision.all import *
+import gradio as gr
+
+try:
+	learn = load_learner("chessman_model.pkl")
+except FileNotFoundError:
+	print("Model file not found. Please ensure 'chessman_model.pkl' exists.")
+	exit(1)
+
+def classify_image(img):
+    # Gradio will pass a PIL.Image if we set type="pil"
+    pred_class, pred_idx, probs = learn.predict(PILImage.create(img))
+    # Return top-3 as a dict for gr.Label
+    topk = probs.topk(3)
+    return {learn.dls.vocab[i]: float(probs[i]) for i in topk.indices}
+
+demo = gr.Interface(
+	fn=classify_image,
+	inputs=gr.Image(type="pil", label="Upload chess piece", height=320),
+	outputs=gr.Label(num_top_classes=3, label="Prediction"),
+	examples=['knight.jpg', 'pawn.JPG', 'rook.jpg'],
+	title="Chess Piece Classifier",
+	description="Upload an image of a chess piece to identify it!"
+)
+
+demo.launch(inline=True, debug=True)
+
+

chessman_model.pkl

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

requirements.txt

@@ -0,0 +1,9 @@
+fastai>=2.7.0
+gradio>=4.0.0
+torch>=2.0.0
+torchvision>=0.15.0
+pillow>=9.0.0
+numpy>=1.21.0
+matplotlib>=3.5.0
+pandas>=1.5.0
+scikit-learn>=1.2.0