Update app.py

app.py

@@ -9,6 +9,12 @@ from huggingface_hub import hf_hub_download
 from imgutils.data import load_image
 from imgutils.utils import open_onnx_model
 
+import gradio as gr
+from huggingface_hub import hf_hub_download
+from onnx import hub
+import onnxruntime as ort
+
+
 _MODELS = [
     ('content_moderation.onnx', 224),
 ]
@@ -16,6 +22,13 @@ _MODEL_NAMES = [name for name, _ in _MODELS]
 _DEFAULT_MODEL_NAME = _MODEL_NAMES[0]
 _MODEL_TO_SIZE = dict(_MODELS)
 
+model = hub.load(hf_hub_download(
+        'tanlocc/Out_of_Universe',
+        f'{name}'
+    ))
+
+print(model)
+
 
 @lru_cache()
 def _onnx_model(name):