clean up

app.py

@@ -19,12 +19,13 @@ def get_tokenizer(model_name):
 
 @st.experimental_singleton
 def get_model(model_name):
-    return MarianMTModel.from_pretrained(model_name).to(device)
+    model = MarianMTModel.from_pretrained(model_name).to(device)
+    print(f"Loaded model, {model.num_parameters():,d} parameters.")
+    return model
 
 tokenizer = get_tokenizer(model_name)
 model = get_model(model_name)
 
-print(f"The model has {model.num_parameters():,d} parameters.")
 
 input_text = st.text_input("Enter text to translate", "Hola, mi nombre es Juan")
 input_text = input_text.strip()
@@ -67,4 +68,3 @@ with tokenizer.as_target_tokenizer():
 
 
 st.write(probs_table)
-st.write(model.config.decoder_start_token_id)