sks01dev commited on
Commit
ce27794
·
1 Parent(s): d567c4d

Deploy model and app with LFS tracking enabled.

Browse files
Files changed (7) hide show
  1. .gitattributes +2 -0
  2. app.py +35 -0
  3. badminton.jpg +3 -0
  4. cricket.jpg +3 -0
  5. model.pkl +3 -0
  6. requirements.txt +5 -0
  7. swimming.jpg +3 -0
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ *.jpg filter=lfs diff=lfs merge=lfs -text
37
+ *.png filter=lfs diff=lfs merge=lfs -text
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import gradio as gr
3
+ from fastai.vision.all import *
4
+ import os
5
+
6
+ # --- Model Loading (Assumes model.pkl exists in the root) ---
7
+ try:
8
+ learn = load_learner('export.pkl')
9
+ except Exception:
10
+ print("Error loading export.pkl. Check file path/existence.")
11
+ raise
12
+
13
+ labels = learn.dls.vocab
14
+
15
+ def predict(img):
16
+ img = PILImage.create(img)
17
+ pred, pred_idx, probs = learn.predict(img)
18
+ return {labels[i]: float(probs[i]) for i in range(len(labels))}
19
+
20
+ # --- Interface Setup ---
21
+ title = "Sports Classifier"
22
+ description = "A sports classifier trained on the images from Google. Created as a demo for Gradio and HuggingFace Spaces."
23
+ article="<p style='text-align: center'><a href='www.linkedin.com/in/shivamkswarnkar' target='_blank'>Linkedin Profile</a></p>"
24
+ enable_queue=True
25
+ examples = ["badminton.jpg", "cricket.jpg", "swimming.jpg"]
26
+ demo = gr.Interface(fn=predict,
27
+ inputs=gr.Image(type="pil"),
28
+ outputs=gr.Label(num_top_classes=3),
29
+ title=title,
30
+ description=description,
31
+ article=article,
32
+ examples=examples)
33
+
34
+ if __name__ == "__main__":
35
+ demo.launch()
badminton.jpg ADDED

Git LFS Details

  • SHA256: 0bc747cefa3bbd4d57c0f61cc3057b9414e7d1d4f2566e99acbee110323528a0
  • Pointer size: 131 Bytes
  • Size of remote file: 198 kB
cricket.jpg ADDED

Git LFS Details

  • SHA256: 5c60b7833ca99ecd4cd61e1f2adb906ad47a32dc90b9fcb9c586709f9789ff4e
  • Pointer size: 131 Bytes
  • Size of remote file: 311 kB
model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b037e5eb7f44f17f0e22a0d9386696c068a7d1b9b0d0cfd4770d3144f22b222
3
+ size 87546221
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+
2
+ gradio
3
+ torch
4
+ scikit-image
5
+ fastai<2.8.0 # older compatible version
swimming.jpg ADDED

Git LFS Details

  • SHA256: 6eac956b6e28469ed32c0af20dadee53fc0901e5e3c5e8b141f4ecabbd5780a0
  • Pointer size: 131 Bytes
  • Size of remote file: 147 kB