Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4# Load model
5tokenizer = AutoTokenizer.from_pretrained("kunaliitkgp09/agriculture-specialist-gpt2")
6model = AutoModelForCausalLM.from_pretrained("kunaliitkgp09/agriculture-specialist-gpt2")
7
8# Set pad token
9if tokenizer.pad_token is None:
10 tokenizer.pad_token = tokenizer.eos_token
11
12# Example usage
13question = "What are the best practices for organic farming?"
14prompt = f"### Instruction:
15{question}
16
17### Response:
18"
19
20inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
21
22with torch.no_grad():
23 outputs = model.generate(
24 **inputs,
25 max_new_tokens=200,
26 temperature=0.8,
27 do_sample=True,
28 top_p=0.9,
29 top_k=50,
30 pad_token_id=tokenizer.eos_token_id,
31 eos_token_id=tokenizer.eos_token_id,
32 repetition_penalty=1.1,
33 )
34
35response = tokenizer.decode(outputs[0], skip_special_tokens=True)
36print(response)1@misc{agriculture-specialist-gpt2,
2 title={Agriculture Specialist GPT-2 Model},
3 author={Kunal Dhanda},
4 year={2024},
5 url={https://huggingface.co/kunaliitkgp09/agriculture-specialist-gpt2},
6 note={Fine-tuned GPT-2 model for agriculture Q&A}
7}