Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
11 |
-
table_class_index =
|
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):
|