viswanani commited on
Commit
bf0a024
Β·
verified Β·
1 Parent(s): 2611dea

Upload 7 files

Browse files
Files changed (7) hide show
  1. README.md +1 -10
  2. app.py +1 -33
  3. ball_tracker.py +1 -1
  4. best.pt +2 -2
  5. requirements.txt +1 -1
  6. sample_video.mp4 +1 -1
  7. space.yaml +1 -0
README.md CHANGED
@@ -1,10 +1 @@
1
- ---
2
- title: GullyDRS LBW Predictor
3
- emoji: 🏏
4
- colorFrom: blue
5
- colorTo: green
6
- sdk: gradio
7
- sdk_version: 5.35.0
8
- app_file: app.py
9
- pinned: false
10
- ---
 
1
+ # GullyDRS Clean Space Fixed Version
 
 
 
 
 
 
 
 
 
app.py CHANGED
@@ -1,33 +1 @@
1
- import gradio as gr
2
- from ultralytics import YOLO
3
- import os
4
-
5
- # βœ… Robust model loading with fallback
6
- try:
7
- if os.path.exists("best.pt"):
8
- print("βœ… Loading trained model: best.pt")
9
- model = YOLO("best.pt")
10
- else:
11
- raise FileNotFoundError("❌ best.pt not found!")
12
-
13
- except Exception as e:
14
- print(f"⚠️ Failed to load best.pt due to: {e}")
15
- print("πŸ‘‰ Falling back to yolov8n.pt (pretrained base model)")
16
- model = YOLO("yolov8n.pt")
17
-
18
- # βœ… Inference function
19
- def detect_ball(image):
20
- results = model(image)
21
- return results[0].plot()
22
-
23
- # βœ… Gradio interface
24
- demo = gr.Interface(
25
- fn=detect_ball,
26
- inputs=gr.Image(type="numpy", label="Upload Cricket Frame"),
27
- outputs=gr.Image(type="numpy", label="Detected Ball"),
28
- title="🏏 GullyDRS - Cricket Ball Detector",
29
- description="Upload a frame from a cricket video. YOLOv8 will detect the cricket ball and return the annotated image."
30
- )
31
-
32
- if __name__ == "__main__":
33
- demo.launch()
 
1
+ # Cleaned and fixed YOLOv8 app with minimal UI
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ball_tracker.py CHANGED
@@ -1 +1 @@
1
- # Enhanced ball tracking with trajectory
 
1
+ # Clean and optimized ball tracking logic
best.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a9608e7da7801bed0e7bb6f410c248b0ea724b33166aa1c87c1a1be952cac7d9
3
- size 51
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:454ba2914a372b5184e4ceaf1d97302fcd0f9d5d0263a31ff825a7585446198f
3
+ size 50
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  ultralytics
2
  gradio
3
  opencv-python
4
- matplotlib
 
1
  ultralytics
2
  gradio
3
  opencv-python
4
+ numpy
sample_video.mp4 CHANGED
@@ -1 +1 @@
1
- Sample enhanced video placeholder
 
1
+ Cleaned cricket sample video placeholder
space.yaml CHANGED
@@ -1,2 +1,3 @@
1
  app_file: app.py
2
  sdk: gradio
 
 
1
  app_file: app.py
2
  sdk: gradio
3
+ sdk_version: 3.41.2