Spaces:
Running
Running
| import numpy as np | |
| from huggingface_hub import hf_hub_url, cached_download | |
| import joblib | |
| REPO_ID = "MalchuL/JJBAGAN" | |
| FILENAME = "198_jjba_8_k_2_099_ep.onnx" | |
| model = cached_download( | |
| hf_hub_url(REPO_ID, FILENAME) | |
| ) | |
| print(model) | |
| import onnxruntime | |
| ort_session = onnxruntime.InferenceSession(str(model)) | |
| input_name = ort_session.get_inputs()[0].name | |
| ort_inputs = {input_name: np.random.randn(1, 3, 256, 256).astype(dtype=np.float32)} | |
| ort_outs = ort_session.run(None, ort_inputs) | |
| print(ort_outs) |