Rename inference.py to handler.py

9278392 verified almost 2 years ago

155 Bytes

from ultralytics import YOLO
import torch

def run_inference(image_path):
    model = YOLO('model.pt')
    results = model(image_path)
    return results


	from ultralytics import YOLO
	import torch

	def run_inference(image_path):
	model = YOLO('model.pt')
	results = model(image_path)
	return results