Skip to content

Commit

Permalink
style
Browse files Browse the repository at this point in the history
  • Loading branch information
MateoLostanlen committed May 20, 2024
1 parent 6bf7bde commit cbabdcf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pyroengine/vision.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def __init__(self, model_path: Optional[str] = "data/model.onnx", img_size: tupl
print(f"Downloading model from {MODEL_URL} ...")
with DownloadProgressBar(unit="B", unit_scale=True, miniters=1, desc=model_path) as t:
urlretrieve(MODEL_URL, model_path, reporthook=t.update_to)

Check warning on line 40 in pyroengine/vision.py

View check run for this annotation

Codacy Production / Codacy Static Code Analysis

pyroengine/vision.py#L40

Audit url open for permitted schemes. Allowing use of file:/ or custom schemes is often unexpected.
print(f"Model downloaded!")
print("Model downloaded!")

self.ort_session = onnxruntime.InferenceSession(model_path)
self.img_size = img_size
Expand Down

0 comments on commit cbabdcf

Please sign in to comment.