Update app.py

app.py

@@ -7,9 +7,14 @@ from ultralytics import YOLO
 # Load the trained model
 
 
+@spaces.GPU
+def model_cov():
+    model = YOLO("best.pt")
+    model.export(format="engine")  # creates 'yolov8n.engine'
 
-model = YOLO("best.pt")
+model_cov()
 
+model =YOLO("best.engine")
 
 # Define the class indices for figures and tables
 figure_class_index = 3  # class index for figures