tanlocc commited on
Commit
6f8f315
Β·
1 Parent(s): 9273c0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ _MODEL_TO_SIZE = dict(_MODELS)
18
 
19
  @lru_cache()
20
  def _onnx_model(name):
21
- return open_onnx_model(f"tanlocc/NSFW_Check/blob/main/content_moderation_opset15.onnx")
22
 
23
 
24
  def _image_preprocess(image, size: int = 224) -> np.ndarray:
 
18
 
19
  @lru_cache()
20
  def _onnx_model(name):
21
+ return open_onnx_model(f"main/content_moderation_opset15.onnx")
22
 
23
 
24
  def _image_preprocess(image, size: int = 224) -> np.ndarray: