Spaces:

foduucom
/

table-extraction-yolov8

Running

nehulagrawal commited on Aug 6, 2023

Commit

c313ec6

•

1 Parent(s): b4278bd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ def yolov8_inference(
     model = YOLO(model_path)
     model.conf = conf_threshold
     model.iou = iou_threshold
-    results = model.predict(image, imgsz=image_size, return_outputs=True)
     object_prediction_list = []
     for _, image_results in enumerate(results):
         if len(image_results)!=0:

     model = YOLO(model_path)
     model.conf = conf_threshold
     model.iou = iou_threshold
+    results = model.predict(image, imgsz=image_size)
     object_prediction_list = []
     for _, image_results in enumerate(results):
         if len(image_results)!=0: