Update app.py

app.py

@@ -6,11 +6,15 @@ from ultralytics import YOLOv10
 
 # Load the trained model
 
+@spaces.GPU
+def model_cv()
+    model = YOLOv10("best.pt")
 
-model = YOLOv10("best.pt")
-
-model.export(format='engine',imgsz=640, iou=0.7, device = 0, simplify=True, half = True, workspace=8)
+    model.export(format='engine',imgsz=640, iou=0.7, device = 0, simplify=True, half = True, workspace=8)
+    model = YOLOv10('yolov10l.engine', task='detect')
+    return model
 
+model = model_cv()
 
 # Define the class indices for figures and tables
 figure_class_index = 3  # class index for figures