Update README.md

README.md

@@ -11,12 +11,27 @@ tags:
 - trl
 ---
 
-# Uploaded  model
+```
+prompt = """You are tasked with writing a poem in the style of the poet mentioned below. The poem should fit the specified category and adhere to the poet's distinctive tone, language, and themes.
 
-- **Developed by:** rishiraj
-- **License:** apache-2.0
-- **Finetuned from model :** unsloth/gemma-2-9b-bnb-4bit
+### Poet:
+{}
 
-This gemma2 model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
+### Category:
+{}
 
-[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
+### Poem:
+{}"""
+
+inputs = tokenizer(
+[
+    prompt.format(
+        "সুনীল গঙ্গোপাধ্যায়",
+        "প্রেমমূলক",
+        "",
+    )
+], return_tensors = "pt").to("cuda")
+
+outputs = model.generate(**inputs, max_new_tokens = 256, use_cache = True)
+tokenizer.batch_decode(outputs)
+```