Manasa1 commited on
Commit
61c4d3d
·
verified ·
1 Parent(s): 9fbf729

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ class TweetDatasetProcessor:
105
  """
106
 
107
  input_ids = self.tokenizer.encode(prompt, return_tensors='pt')
108
- output = self.model.generate(input_ids, max_length=150, num_return_sequences=1, temperature=1.0)
109
  generated_tweet = self.tokenizer.decode(output[0], skip_special_tokens=True).strip()
110
 
111
  return generated_tweet
 
105
  """
106
 
107
  input_ids = self.tokenizer.encode(prompt, return_tensors='pt')
108
+ output = self.model.generate(input_ids, max_length=150,max_new_token = 300, num_return_sequences=1, temperature=1.0)
109
  generated_tweet = self.tokenizer.decode(output[0], skip_special_tokens=True).strip()
110
 
111
  return generated_tweet