Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_name = "Zeeshan506/echohire-qgen-distilgpt2"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)
6
7prompt = "Job Title: Backend Developer\nSkills: Python, FastAPI, PostgreSQL\nQuestions:"
8
9inputs = tokenizer(prompt, return_tensors="pt")
10input_ids = inputs["input_ids"]
11attention_mask = (input_ids != tokenizer.pad_token_id).long()
12
13output_ids = model.generate(
14 input_ids=input_ids,
15 attention_mask=attention_mask,
16 max_new_tokens=300,
17 num_beams=3,
18 no_repeat_ngram_size=2
19)
20
21generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
22print(generated_text)@misc{shah2025echohire,
title={EchoHire Question Generator (DistilGPT-2 Fine-tuned)},
author={Shah, Syed Zeeshan},
year={2025},
howpublished={\url{https://huggingface.co/Zeeshan506/echohire-qgen-distilgpt2}}
}