Upload detect.py

detect.py

@@ -0,0 +1,10 @@
+import torch
+
+def load_model(model_path):
+    model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path)
+    return model
+
+def detect(image, model):
+    results = model(image)
+    return results
+