Update README.md

README.md

@@ -34,14 +34,19 @@ model, tokenizer = (AutoModelForCausalLM.from_pretrained("Abhishekcr448/Tiny-Hin
 # Function to generate text
 def generate_text(prompt):
     inputs = tokenizer(prompt, return_tensors='pt').to(model.device)
-    print(tokenizer.decode(model.generate(inputs['input_ids'], 
-                                          max_length=inputs['input_ids'].shape[-1] + 25, 
-                                          no_repeat_ngram_size=2, temperature=0.8, 
-                                          top_k=50, top_p=0.9, do_sample=True)[0], 
-                           skip_special_tokens=True))
-
-while (prompt := input("Enter prompt ('exit' to quit): ").lower()) != "exit":
-    generate_text(prompt)
+    output = model.generate(inputs['input_ids'], 
+                            max_length=inputs['input_ids'].shape[-1] + 25, 
+                            no_repeat_ngram_size=2, temperature=0.8, 
+                            top_k=50, top_p=0.9, do_sample=True)
+    return tokenizer.decode(output[0], skip_special_tokens=True)
+
+# Hardcoded input for testing
+prompt = 'kya kar rahe ho'
+print(generate_text(prompt))
+```
+**Output**
+```output
+kya kar rahe ho bas yaar ghar pe hi hoon main toh kabhi kabhi itna kaam karta hoon ki bore ho jata hoon
 ```
 
 ## Model Details