from inference import predict, reset

# Test predict
print("Testing predict...")
output = predict({"query": "I like AI"})
print(output)

# Test reset
print("Testing reset...")
print(reset())