linhcuem commited on
Commit
97afcf2
1 Parent(s): f3958f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # model = YOLO('linhcuem/chamdiem_yolov8_ver10')
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(