Edit model card

Model Card for Agro-QA

This model is fine-tuned for agricultural question-answering tasks. It leverages the Llama-3.2-3B-Instruct model to address a variety of topics in agriculture, such as crop selection, pest management, irrigation, and farming best practices.

Model Details

Model Description

  • Developed by: Shukla Shreyansh
  • Model type: Question Answering (QA)
  • Language(s) (NLP): English
  • License: Apache-2.0
  • Finetuned from model: unsloth/Llama-3.2-3B-Instruct

Uses

Direct Use

The model is intended for question-answering applications specific to agriculture. It provides insights into farming techniques, crop choices, pest management, and related topics.

Out-of-Scope Use

The model is not designed for non-agriculture-related questions or tasks requiring specialized domain knowledge outside of agriculture.


Training Details

Training Data

The model is fine-tuned on the KisanVaani/agriculture-qa-english-only dataset, a curated collection of questions and answers focused on agricultural topics.

Training Procedure

  • Training regime: Mixed precision (FP16)
  • Batch size: 2 (per device)
  • Epochs: 1
  • Learning rate: 2e-4
  • Optimizer: AdamW with 8-bit precision

Evaluation

Testing Data

The model is evaluated on a subset of the training dataset to measure its performance in answering agriculture-related questions.

Metrics

  • Accuracy: [More Information Needed]
  • F1 Score: [More Information Needed]

How to Get Started with the Model

Use the code below to load and use the model:

from transformers import AutoTokenizer, AutoModelForCausalLM

# Load tokenizer
tokenizer = AutoTokenizer.from_pretrained("ShuklaShreyansh/Agro-QA")

# Load model
model = AutoModelForCausalLM.from_pretrained("ShuklaShreyansh/Agro-QA").to("cuda")

# Example usage
messages = [{"role": "user", "content": "What are the best rabi crops to grow?"}]
inputs = tokenizer.apply_chat_template(messages, tokenize=True, return_tensors="pt").to("cuda")
output = model.generate(input_ids=inputs['input_ids'], max_new_tokens=128)
print(tokenizer.decode(output[0]))

Model Card for Model ID

This modelcard aims to be a base template for new models. It has been generated using this raw template.

Model Details

Model Description

  • Developed by: [More Information Needed]
  • Funded by [optional]: [More Information Needed]
  • Shared by [optional]: [More Information Needed]
  • Model type: [More Information Needed]
  • Language(s) (NLP): [More Information Needed]
  • License: [More Information Needed]
  • Finetuned from model [optional]: [More Information Needed]

Model Sources [optional]

  • Repository: [More Information Needed]
  • Paper [optional]: [More Information Needed]
  • Demo [optional]: [More Information Needed]

Uses

Direct Use

[More Information Needed]

Downstream Use [optional]

[More Information Needed]

Out-of-Scope Use

[More Information Needed]

Bias, Risks, and Limitations

[More Information Needed]

Recommendations

Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.

How to Get Started with the Model

Use the code below to get started with the model.

[More Information Needed]

Training Details

Training Data

[More Information Needed]

Training Procedure

Preprocessing [optional]

[More Information Needed]

Training Hyperparameters

  • Training regime: [More Information Needed]

Speeds, Sizes, Times [optional]

[More Information Needed]

Evaluation

Testing Data, Factors & Metrics

Testing Data

[More Information Needed]

Factors

[More Information Needed]

Metrics

[More Information Needed]

Results

[More Information Needed]

Summary

Model Examination [optional]

[More Information Needed]

Environmental Impact

Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).

  • Hardware Type: [More Information Needed]
  • Hours used: [More Information Needed]
  • Cloud Provider: [More Information Needed]
  • Compute Region: [More Information Needed]
  • Carbon Emitted: [More Information Needed]

Technical Specifications [optional]

Model Architecture and Objective

[More Information Needed]

Compute Infrastructure

[More Information Needed]

Hardware

[More Information Needed]

Software

[More Information Needed]

Citation [optional]

BibTeX:

[More Information Needed]

APA:

[More Information Needed]

Glossary [optional]

[More Information Needed]

More Information [optional]

[More Information Needed]

Model Card Authors [optional]

[More Information Needed]

Model Card Contact

[More Information Needed]

Downloads last month
22
Safetensors
Model size
1.85B params
Tensor type
F32
·
FP16
·
U8
·
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.

Model tree for ShuklaShreyansh/Agro-QA

Finetuned
(69)
this model

Dataset used to train ShuklaShreyansh/Agro-QA