Spaces:

zliang
/

fastpaperlayout

Sleeping

zliang commited on May 25

Commit

57d20a1

•

1 Parent(s): 7aaeeb7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,12 +2,12 @@ import gradio as gr
 import numpy as np
 import fitz  # PyMuPDF
 import spaces
-from ultralytics import YOLO
 # Load the trained model
-model = YOLO("best.engine")

 import numpy as np
 import fitz  # PyMuPDF
 import spaces
+from ultralytics import YOLOv10
 # Load the trained model
+model = YOLOv10("best.pt")