Update app.py

app.py

@@ -6,17 +6,15 @@ import spaces
 
 # Load the trained model
 
-# Load the YOLOv8 model
-pt_model = YOLO("best.pt")
 
-# Export the model to TensorRT format
 @spaces.GPU
 def model_c():
-    pt_model.export(format="engine")  # creates 'yolov8n.engine'
+    model = YOLO("best.pt")
+
 
 model_c()
 # Load the exported TensorRT model
-model = YOLO("best.engine")
+
 
 # Define the class indices for figures and tables
 figure_class_index = 3  # class index for figures