jayanthapoojary1989 commited on
Commit
15d4bb4
·
verified ·
1 Parent(s): 884d57c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import cv2
4
  import numpy as np
5
 
6
  # Load your trained YOLOv8 model
7
- model = YOLO('best.pt') # Update this path to your model weights file
8
 
9
  def detect_pneumonia(image):
10
  # YOLOv8 expects image in numpy array format (BGR)
 
4
  import numpy as np
5
 
6
  # Load your trained YOLOv8 model
7
+ model = YOLO('pytorch_yolov8_model.pt') # Update this path to your model weights file
8
 
9
  def detect_pneumonia(image):
10
  # YOLOv8 expects image in numpy array format (BGR)