zliang commited on
Commit
a88e01c
·
verified ·
1 Parent(s): c65777e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import fitz # PyMuPDF
7
  from PIL import Image
8
 
9
  # Load the trained model
10
- model_path = 'runs/detect/train7/weights/best.pt' # Replace with the path to your trained .pt file
11
  model = YOLO(model_path)
12
 
13
  # Define the class indices for figures and tables
 
7
  from PIL import Image
8
 
9
  # Load the trained model
10
+ model_path = 'best.pt' # Replace with the path to your trained .pt file
11
  model = YOLO(model_path)
12
 
13
  # Define the class indices for figures and tables