Spaces:

zliang
/

fastpaperlayout

Sleeping

zliang commited on May 25

Commit

041f53d

•

1 Parent(s): b58fcaa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,13 +2,13 @@ import gradio as gr
 import numpy as np
 import fitz  # PyMuPDF
 import spaces
-from ultralytics import YOLOv10
 # Load the trained model
-model = YOLOv10("best.pt")
 # Define the class indices for figures and tables

 import numpy as np
 import fitz  # PyMuPDF
 import spaces
+from ultralytics import YOLO
 # Load the trained model
+model = YOLO("best.pt")
 # Define the class indices for figures and tables