syedfaisalabrar commited on
Commit
61a57f9
·
verified ·
1 Parent(s): 6f00f3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from ultralytics import YOLO # Import YOLO from Ultralytics
3
  from PIL import Image
4
  import numpy as np
5
 
6
- # Load your YOLO model (Make sure the model file is in the same directory as this script)
7
- model = YOLO('yolov8s.pt') # Use the YOLO class to load the model
8
 
9
  # Define the prediction function for YOLO model
10
  def predict(image):
 
3
  from PIL import Image
4
  import numpy as np
5
 
6
+
7
+ model = YOLO('mouseFeetDetection.pt') # Use the YOLO class to load the model
8
 
9
  # Define the prediction function for YOLO model
10
  def predict(image):