kostya-cholak commited on
Commit
41c19be
·
1 Parent(s): 9c0d975

fix: add .pt extension

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -23,6 +23,8 @@ def load_model():
23
  new_cached_model_path = f"{cached_model_path}.pt"
24
  os.rename(cached_model_path, new_cached_model_path)
25
 
 
 
26
  # Load the model using YOLO from the cached model file
27
  return YOLO(new_cached_model_path)
28
 
 
23
  new_cached_model_path = f"{cached_model_path}.pt"
24
  os.rename(cached_model_path, new_cached_model_path)
25
 
26
+ print(f"Downloaded model to {new_cached_model_path}")
27
+
28
  # Load the model using YOLO from the cached model file
29
  return YOLO(new_cached_model_path)
30