Update app.py

app.py

@@ -31,6 +31,7 @@ mdl_kwargs = dict(
 )
 if MODEL_REV: mdl_kwargs["revision"] = MODEL_REV
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID, **mdl_kwargs).to(device)
+model.config.vocab_size = model.get_output_embeddings().weight.shape[0]
 model.eval()
 
 ip = IndicProcessor(inference=True)