zliang commited on
Commit
4f91707
·
verified ·
1 Parent(s): a2d0f56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from ultralytics import YOLO
3
  import numpy as np
4
  import fitz # PyMuPDF
5
  import spaces
@@ -9,7 +9,7 @@ import spaces
9
 
10
 
11
 
12
- model = YOLO("best.pt")
13
 
14
 
15
  # Define the class indices for figures and tables
 
1
  import gradio as gr
2
+ from ultralytics import YOLOv10
3
  import numpy as np
4
  import fitz # PyMuPDF
5
  import spaces
 
9
 
10
 
11
 
12
+ model = YOLOv10("best.pt")
13
 
14
 
15
  # Define the class indices for figures and tables