File size: 338 Bytes
092415c |
1 |
"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)" |