zliang commited on
Commit
414fdff
1 Parent(s): 8bbef17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ import re
20
 
21
 
22
  # Load the trained model
23
- model = YOLO("runs\\detect\\train7\\weights\\best.pt")
24
  openai_api_key = os.environ.get("openai_api_key")
25
 
26
  # Define the class indices for figures, tables, and text
 
20
 
21
 
22
  # Load the trained model
23
+ model = YOLO("best.pt")
24
  openai_api_key = os.environ.get("openai_api_key")
25
 
26
  # Define the class indices for figures, tables, and text