Edit model card

AgriBrain's AI-core, agbrain


AbriBrain's AI-core, agbrain, is a cutting-edge natural language processing (NLP) model built specifically for generating content related to agriculture. The model is a fine-tuned version of the popular GPT-2 language model, trained on a vast corpus of 1601 PDF documents sourced from various reputable online resources.

Agbrain has been specifically designed to cater to the needs of the agriculture industry, including farmers, agronomists, agricultural researchers, and other stakeholders.

One of the key strengths of Agbrain is its ability to generate coherent, and contextually relevant content. The model has been fine-tuned using advanced machine learning techniques to ensure that the generated content is both accurate and informative. It is capable of producing content on a wide range of topics, including crop cultivation, livestock management, pest control, irrigation, and more.

Overall, Agbrain is a powerful and versatile NLP model that is perfectly suited to the needs of the agriculture industry.

Usage


Transformers and model.generate


import tensorflow as tf
from transformers import TFGPT2LMHeadModel, GPT2Tokenizer

tokenizer = GPT2Tokenizer.from_pretrained("benkimz/agbrain")
model = TFGPT2LMHeadModel.from_pretrained("benkimz/agbrain")

prompt = """
I think agribusiness is a great opportunity for passionate
investors. From food business to growing crops for sale,
and rearing livestock for business.
"""

input_ids = tokenizer.encode(prompt, return_tensors="tf")
outputs = model.generate(input_ids=input_ids,
          max_length=120,
          do_sample=True)
generated_text = tokenizer.decode(outputs[0],
          skip_special_tokens=True)

print(generated_text)

# Output
"""
I think agribusiness is a great opportunity for passionate
investors. From food business to growing crops for sale,
and rearing livestock for business.

In this paper I will introduce a concept model agribusiness
that focuses on businesses to grow large amounts of product.
 This model requires that product be sold outside of
agriculture industry, thus allowing farmers advantages,
especially over agronomic competition in production.
model is very important to farmers as it will be possible,
to sell their products at local markets without 
"""

Transformers pipeline


from transformers import pipeline, set_seed
generator = pipeline('text-generation', model='benkimz/agbrain')
set_seed(42)

samples = generator(
    "Animal husbandry is an important part of livestock production.", 
    max_length=100, 
    num_return_sequences=2
)

for sample in samples:
  print("Model output: {}\n".format(sample['generated_text']))


# Output
"""
**Model output**: Animal husbandry is an important part of
livestock production.  livestock production industry is complex,
many factors contribute to this complexity.  need to determine
most efficient method of handling livestock to ensure best quality
product. It is important that animals being handled appropriately
have properly cleaned equipment that prevents scratching
(Sappell 2002). Because livestock is an important part of
livestock production, veterinary care must be taken regularly
during transport of animals from a farm to your home to be
successful. If livestock were to be

**Model output**: Animal husbandry is an important part of
livestock production. Animal husbandry combines various
strategies to control pests. Management strategies of pest
management strategies
Preventing pest from reaching level
 Preventing pest from reaching level
To minimize transmission costs, control mechanisms
 must be developed to prevent pest from reaching level. In
order to have an accurate information about pest
management methods, instrumental field study of pest management
measures be developed by field of study. A technique of this
"""

Metrics


Step Training Loss
500 3.877700
1000 3.746200
1500 3.659600
2000 3.613300
2500 3.603400
3000 3.561600
3500 3.558300
4000 3.518400
4500 3.504100
5000 3.508600

Further training could improve the model and make it better.

Downloads last month
42
Safetensors
Model size
137M params
Tensor type
F32
·
BOOL
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Space using benkimz/agbrain 1