Amodit commited on
Commit
aa54535
·
1 Parent(s): b5c84c6

Fix requirements and updated app model

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -7,8 +7,9 @@ import tempfile
7
  import os
8
 
9
  # Load the model
10
- # yolov8n-pose.pt will be automatically downloaded if not present
11
- model = YOLO('yolov8n-pose.pt')
 
12
 
13
  def get_angle(p1, p2, p3):
14
  """
 
7
  import os
8
 
9
  # Load the model
10
+ # Load the model
11
+ # Using custom model provided by user
12
+ model = YOLO('yolo26x-pose.pt')
13
 
14
  def get_angle(p1, p2, p3):
15
  """
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  ultralytics
2
  gradio
3
- opencv-python
4
  numpy
 
1
  ultralytics
2
  gradio
3
+ opencv-python-headless
4
  numpy