Spaces:

praysimanjuntak
/

extract-solar-data

Sleeping

praysimanjuntak commited on Jul 10

Commit

3cb12d8

•

1 Parent(s): 1d80da5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ import cv2
 import gradio as gr
 # Initialize YOLO model
-model_yolo = YOLO(model="runs/detect/train/weights/best.pt")
 ocr = PaddleOCR(use_angle_cls=True, lang='en', use_gpu=False, show_log=False)
 def process_pdf(file):

 import gradio as gr
 # Initialize YOLO model
+model_yolo = YOLO(model="yolov8n-box.pt")
 ocr = PaddleOCR(use_angle_cls=True, lang='en', use_gpu=False, show_log=False)
 def process_pdf(file):