warvan-ml-model / example.py
warvan's picture
Create example.py
092415c verified
raw
history blame contribute delete
338 Bytes
"example.py": "import torch\nfrom transformers import AutoModel, AutoTokenizer\n\nmodel_name = 'warvan/warvan-ml-model'\ntokenizer = AutoTokenizer.from_pretrained(model_name)\nmodel = AutoModel.from_pretrained(model_name)\n\ntext = 'Hello, world!'\ninputs = tokenizer(text, return_tensors='pt')\noutputs = model(**inputs)\nprint(outputs)"