Spaces:

b2bomber
/

AIWaterMarkRemover

Running

b2bomber commited on Jul 18

Commit

5dc89bc

verified ·

1 Parent(s): b3e54a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,9 +6,9 @@ import os
 import tempfile
 import subprocess
-from ultralytics import YOLO
-model = YOLO("watermark-detection-73.pt")
 model.conf = 0.25
 model.iou = 0.45
 model.max_det = 1

 import tempfile
 import subprocess
+import torch
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='watermark-detection-73.pt', force_reload=True)
 model.conf = 0.25
 model.iou = 0.45
 model.max_det = 1