Spaces:

mf666
/

shit-checker

Runtime error

wtf741 commited on Jan 2

Commit

85cd204

•

1 Parent(s): 8cb8f64

Update check.py

Files changed (1) hide show

check.py CHANGED Viewed

@@ -18,7 +18,7 @@ MODELS = natsorted([
     os.path.splitext(os.path.relpath(file, REPOSITORY))[0]
     for file in hf_fs.glob(f'{REPOSITORY}/*.onnx')
 ])
-DEFAULT_MODEL = 'mobilenet.small'
 @lru_cache()

     os.path.splitext(os.path.relpath(file, REPOSITORY))[0]
     for file in hf_fs.glob(f'{REPOSITORY}/*.onnx')
 ])
+DEFAULT_MODEL = 'mobilenet.xs.v2'
 @lru_cache()