Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|