Update app.py

app.py

@@ -104,8 +104,12 @@ class TweetDatasetProcessor:
         **Only generate the tweet. Do not include analysis, explanation, or any other content.**
         """
 
-        input_ids = self.tokenizer.encode(prompt, return_tensors='pt')
-        output = self.model.generate(input_ids, max_length=150,max_new_token = 300, num_return_sequences=1,do_sample = True,top_k=50,pad_token_id=tokenizer.eos_token_id, temperature=1.0)
+        inputs = self.tokenizer(prompt, return_tensors='pt', truncation=True, padding=True)
+        input_ids = inputs['input_ids']
+        attention_mask = inputs['attention_mask']
+
+        pad_token_id = self.tokenizer.eos_token_id
+        output = self.model.generate(input_ids, attention_mask=attention_mask, max_length=150, num_return_sequences=1, temperature=1.0, pad_token_id=pad_token_id)
         generated_tweet = self.tokenizer.decode(output[0], skip_special_tokens=True).strip()
         
         return generated_tweet