Hugging Face's logo

Spaces:
The-Infinite-Tsukuyomi
/
Smoking-Detection
Runtime error

App Files Community
itachi-ai commited on
Commit
05d0158
·
verified ·
1 Parent(s): 9af4400

change model

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -13,4 +13,4 @@ model = keras_cv.models.YOLOV8Detector(
13
  fpn_depth=2,
14
  )
15
 
16
- model.load_weights('models/model_e15.weights.h5')
 
13
  fpn_depth=2,
14
  )
15
 
16
+ model.load_weights('models/model.weights.h5')