Update app.py

app.py

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