Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_name = "felixmanojh/DJ-AI-Radio"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
7model = AutoModelForCausalLM.from_pretrained(
8 model_name,
9 trust_remote_code=True,
10 torch_dtype=torch.float16
11)
12
13messages = [
14 {"role": "system", "content": "You are a Tamil AI radio DJ who speaks energetic Tanglish. Create engaging commentary."},
15 {"role": "user", "content": "Hype up a high-energy dance track for weekend party"}
16]
17
18text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
19inputs = tokenizer(text, return_tensors="pt")
20
21outputs = model.generate(
22 **inputs,
23 max_new_tokens=150,
24 temperature=0.8,
25 top_p=0.9,
26 do_sample=True
27)
28
29response = tokenizer.decode(outputs[0], skip_special_tokens=True)
30print(response)1@misc{tamil-ai-dj-radio,
2 author = {Felix Manojh},
3 title = {Tamil AI DJ Radio - Tanglish Radio DJ Commentary Generator},
4 year = {2026},
5 publisher = {HuggingFace},
6 url = {https://huggingface.co/felixmanojh/DJ-AI-Radio}
7}