Update app.py

app.py

@@ -62,16 +62,17 @@ trainer = Trainer(
 # Fine-tune the model
 trainer.train()
 
-prompt = "मस्तवाल हत्ती चाळवणं"
+model.save_pretrained("./fine-tuned-mt5-marathi-proverbs")
+tokenizer.save_pretrained("./fine-tuned-mt5-marathi-proverbs")
+
+prompt = st.text_input("Enter your proverb: ")
 
 # Tokenize the input prompt
 input_ids = tokenizer.encode(prompt, return_tensors='pt')
 
 # Generate the output
-output_ids = model.generate(input_ids, max_length=512)
+output_ids = model.generate(input_ids, max_length=256)
 
 # Decode the output to text
-topic = st.text_input("Enter your proverb: ")
 output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
 st.write(output_text)
-