zliang commited on
Commit
ea76efb
1 Parent(s): 028ddc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import spaces
7
  model = YOLOv10("best.pt")
8
 
9
  # Define the class indices for figures and tables
10
- figure_class_index = 3 # class index for figures
11
- table_class_index = 4 # class index for tables
12
 
13
  # Function to perform inference on an image and return bounding boxes for figures and tables
14
  def infer_image_and_get_boxes(image, confidence_threshold=0.6):
 
7
  model = YOLOv10("best.pt")
8
 
9
  # Define the class indices for figures and tables
10
+ figure_class_index = 4 # class index for figures
11
+ table_class_index = 3 # class index for tables
12
 
13
  # Function to perform inference on an image and return bounding boxes for figures and tables
14
  def infer_image_and_get_boxes(image, confidence_threshold=0.6):