Update inference.py

inference.py

@@ -23,8 +23,8 @@ 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),early_stopping=True, max_length=self.max_length,num_beams=10, 
-    no_repeat_ngram_size=4 )
+    generated_ids = self.encoder_decoder_model.generate(pixel_values.unsqueeze(0).to(self.device),early_stopping=True, max_length=self.max_length,num_beams=4, 
+    no_repeat_ngram_size=2 )
     generated_text = self.tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
     return generated_text