Update app.py

app.py

@@ -132,11 +132,12 @@ Please provide a detailed analysis including:
             inputs = self.tiny_tokenizer(prompt, return_tensors="pt", truncation=True)
             outputs = self.tiny_model.generate(
                 inputs["input_ids"],
-                max_length=2048,
+                max_new_tokens=1024,  # Generate up to 1024 new tokens
                 temperature=0.7,
                 top_p=0.95,
-                do_sample=True
-            )
+                do_sample=True,
+                pad_token_id=self.tiny_tokenizer.eos_token_id,
+                repetition_penalty=1.2 )
             analysis = self.tiny_tokenizer.decode(outputs[0], skip_special_tokens=True)
 
             # Generate sentiment