Update inference.py

inference.py

@@ -23,7 +23,7 @@ class Inference:
       self.tokenizer.add_special_tokens({'pad_token': '[PAD]'})
       self.encoder_decoder_model.decoder.resize_token_embeddings(len(self.tokenizer))
 
-    generated_ids = self.encoder_decoder_model.generate(pixel_values.unsqueeze(0).to(self.device), max_length=self.max_length)
+    generated_ids = self.encoder_decoder_model.generate(pixel_values.unsqueeze(0).to(self.device), max_length=self.max_length, early_stopping=True)
     generated_text = self.tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
     return generated_text