Update app.py

app.py

@@ -1,6 +1,4 @@
-import os
-os.system("pip install -q git+https://github.com/THU-MIG/yolov10.git")
-os.system("sed -i 's/config.max_workspace_size = self.args.workspace \* 1 << 30/config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, self.args.workspace \* 1 << 30)/' exporter.py")
+
 import gradio as gr
 import numpy as np
 import fitz  # PyMuPDF
@@ -9,21 +7,15 @@ from ultralytics import YOLOv10
 
 # Load the trained model
 
-@spaces.GPU
-def model_cv():
-    model = YOLOv10("best.pt")
-
-    model.export(format='engine',imgsz=640, iou=0.7, device = 0, simplify=True, half = True)
-    model = YOLOv10('yolov10l.engine', task='detect')
-    return model
+model = YOLOv10("best.pt")
 
-model = model_cv()
 
 # Define the class indices for figures and tables
 figure_class_index = 3  # class index for figures
 table_class_index = 4   # class index for tables
 
 # Function to perform inference on an image and return bounding boxes for figures and tables
+@spaces.GPU
 def infer_image_and_get_boxes(image, confidence_threshold=0.6):
     results = model.predict(image)
     boxes = [