Spaces:
Paused
Paused
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):
|