File size: 1,205 Bytes
ae54537 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
```python
from transformers import GPT2Tokenizer, GPT2LMHeadModel
def generate_response(input_text):
inputs = tokenizer(input_text, return_tensors="pt")
output_sequences = model.generate(
input_ids=inputs['input_ids'],
attention_mask=inputs['attention_mask'],
max_length=100, # Adjusted max_length
temperature=0.3,
top_k=40,
top_p=0.85,
num_return_sequences=1,
no_repeat_ngram_size=2,
pad_token_id=tokenizer.eos_token_id,
early_stopping=True,
do_sample=True,
use_cache=True,
)
full_generated_text = tokenizer.decode(output_sequences[0], skip_special_tokens=True)
bot_response_start = full_generated_text.find('[Bot]') + len('[Bot]')
bot_response = full_generated_text[bot_response_start:]
last_period_index = bot_response.rfind('.')
if last_period_index != -1:
bot_response = bot_response[:last_period_index + 1]
return bot_response.strip()
model_name = 'KhantKyaw/Chat_GPT-2'
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
model = GPT2LMHeadModel.from_pretrained(model_name)
response = generate_response(user_input)
print("Chatbot:", response)
``` |