Update app.py

app.py

@@ -62,3 +62,16 @@ trainer = Trainer(
 # Fine-tune the model
 trainer.train()
 
+prompt = "मस्तवाल हत्ती चाळवणं"
+
+# Tokenize the input prompt
+input_ids = tokenizer.encode(prompt, return_tensors='pt')
+
+# Generate the output
+output_ids = model.generate(input_ids, max_length=512)
+
+# 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)
+