Views
No views yet
HuggingFaceTB/SmolLM2-135M-Instruct1from transformers import AutoTokenizer, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained("PraneelChetty-1/smollm2-135m-query-expansion")
4pipe = pipeline("text-generation", model="PraneelChetty-1/smollm2-135m-query-expansion", device_map="auto")
5
6messages = [{"role": "user", "content": "yield monitoring systems"}]
7prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
8
9out = pipe(prompt, max_new_tokens=80, do_sample=True, temperature=0.7)
10print(out[0]["generated_text"][len(prompt):])