Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
import torch
|
3 |
from sahi.prediction import ObjectPrediction
|
4 |
from sahi.utils.cv import visualize_object_predictions, read_image
|
5 |
-
from ultralyticsplus import YOLO
|
6 |
|
7 |
# from ultralyticsplus import render_result
|
8 |
# import requests
|
@@ -19,7 +19,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','linhcuem/chamdiem_yolov8_
|
|
19 |
['test_images/ee106392e56837366e79.jpg','linhcuem/chamdiem_yolov8_ver10', 640, 0.25, 0.45], ['test_images/f88d2214a4ee76b02fff.jpg','linhcuem/chamdiem_yolov8_ver10', 640, 0.25, 0.45]]
|
20 |
|
21 |
# Load YOLO model
|
22 |
-
|
23 |
|
24 |
###################################################
|
25 |
def yolov8_img_inference(
|
|
|
2 |
import torch
|
3 |
from sahi.prediction import ObjectPrediction
|
4 |
from sahi.utils.cv import visualize_object_predictions, read_image
|
5 |
+
from ultralyticsplus import YOLO, render_result
|
6 |
|
7 |
# from ultralyticsplus import render_result
|
8 |
# import requests
|
|
|
19 |
['test_images/ee106392e56837366e79.jpg','linhcuem/chamdiem_yolov8_ver10', 640, 0.25, 0.45], ['test_images/f88d2214a4ee76b02fff.jpg','linhcuem/chamdiem_yolov8_ver10', 640, 0.25, 0.45]]
|
20 |
|
21 |
# Load YOLO model
|
22 |
+
model = YOLO('linhcuem/chamdiem_yolov8_ver10')
|
23 |
|
24 |
###################################################
|
25 |
def yolov8_img_inference(
|