,,,python def generate_title(keywords): input_ids = tokenizer(keywords, return_tensors="pt", padding="longest", truncation=True, max_length=32).input_ids.to(device) outputs = model.generate( input_ids, num_beams=3, num_beam_groups=3, num_return_sequences=3, repetition_penalty=7.0, diversity_penalty=4.0, no_repeat_ngram_size=3, temperature=0.9, max_length=32 ) return tokenizer.batch_decode(outputs, skip_special_tokens=True)
keywords = 'This repository contains a fine-tuned model for generating high-quality product descriptions.' generate_title(keywords) ,,,
- Downloads last month
- 10
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.