README.md

metadata

license: apache-2.0
language:
  - en
library_name: transformers

Model Name

This is a chatbot model trained using a transformer architecture. It can answer questions based on the training data provided.

Usage

from transformers import AutoTokenizer, TFAutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("your_username/your_model_name")
model = TFAutoModelForSeq2SeqLM.from_pretrained("your_username/your_model_name")

def predict(sentence):
    inputs = tokenizer(sentence, return_tensors="tf")
    outputs = model.generate(inputs["input_ids"])
    return tokenizer.decode(outputs[0], skip_special_tokens=True)

sentence = "Can exposure to pesticides in food impact cognitive health?"
print(predict(sentence))