Manasa1 commited on
Commit
b92a169
·
verified ·
1 Parent(s): 7dfa9a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -109,7 +109,7 @@ class TweetDatasetProcessor:
109
  attention_mask = inputs['attention_mask']
110
 
111
  pad_token_id = self.tokenizer.eos_token_id
112
- output = self.model.generate(input_ids, attention_mask=attention_mask, max_length=500, num_return_sequences=1, temperature=1.0, pad_token_id=pad_token_id)
113
  generated_tweet = self.tokenizer.decode(output[0], skip_special_tokens=True).strip()
114
 
115
  return generated_tweet
 
109
  attention_mask = inputs['attention_mask']
110
 
111
  pad_token_id = self.tokenizer.eos_token_id
112
+ output = self.model.generate(input_ids, attention_mask=attention_mask, max_length=500, num_return_sequences=1, temperature=1.0, pad_token_id=pad_token_id,do_sample = True)
113
  generated_tweet = self.tokenizer.decode(output[0], skip_special_tokens=True).strip()
114
 
115
  return generated_tweet