Update app.py

app.py

@@ -9,7 +9,7 @@ from imgutils.data import load_image
 from imgutils.utils import open_onnx_model
 
 _MODELS = [
-    ('content_moderation_opset15.onnx', 224),
+    ('content_moderation.onnx', 224),
 ]
 _MODEL_NAMES = [name for name, _ in _MODELS]
 _DEFAULT_MODEL_NAME = _MODEL_NAMES[0]
@@ -18,8 +18,10 @@ _MODEL_TO_SIZE = dict(_MODELS)
 
 @lru_cache()
 def _onnx_model(name):
-    return open_onnx_model(f"main/content_moderation_opset15.onnx")
-
+    return open_onnx_model(hf_hub_download(
+        'tanlocc/Out_of_Universe',
+        f'{name}'
+    ))
 
 def _image_preprocess(image, size: int = 224) -> np.ndarray:
     image = load_image(image, mode='RGB').resize((size, size), Image.NEAREST)