Spaces:

stphtan94117
/

TW_plate_number

Sleeping

stphtan94117 commited on Jun 11, 2023

Commit

935485e

1 Parent(s): 125064a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def yolov5_inference(
     iou_threshold: gr.inputs.Slider = 0.45,
 ):
-    model = torch.hub.load('ultralytics/yolov5', 'custom', path='plate.pt', source="local")
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)

     iou_threshold: gr.inputs.Slider = 0.45,
 ):
+    model = torch.hub.load('yolov5', 'custom', path='plate.pt', source="local")
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)