Rammadaeus's picture
Upload fuzz/runner.py with huggingface_hub
7310b92 verified
import tensorflow as tf
import numpy as np
import sys
try:
interp = tf.lite.Interpreter(model_path=sys.argv[1])
interp.allocate_tensors()
for d in interp.get_input_details():
interp.set_tensor(d[index], np.zeros(d[shape], dtype=d[dtype]))
interp.invoke()
print(EXECUTED)
except Exception as e:
print(type(e).__name__)