Views
No views yet
1from transformers import AutoTokenizer, TFAutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained("your_username/your_model_name")
4model = TFAutoModelForSeq2SeqLM.from_pretrained("your_username/your_model_name")
5
6def predict(sentence):
7 inputs = tokenizer(sentence, return_tensors="tf")
8 outputs = model.generate(inputs["input_ids"])
9 return tokenizer.decode(outputs[0], skip_special_tokens=True)
10
11sentence = "Can exposure to pesticides in food impact cognitive health?"
12print(predict(sentence))