Spaces:
Build error
Build error
File size: 643 Bytes
6e4fc64 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
#!/usr/bin/env python
import argparse
from transformers import pipeline
from transformers import AutoModelForSeq2SeqLM
from transformers import AutoTokenizer
def get_args():
""" args from input
"""
parser = argparse.ArgumentParser(description='HSIC-Bottleneck research')
parser.add_argument('-ipt', '--input', required=True,
type=str, help='input path')
args = parser.parse_args()
return args
def main():
args = get_args()
input_file = args.input
with open(input_file, 'r') as file:
data = file.read().rstrip()
print(data)
if __name__ == '__main__':
main() |