Update README.md

README.md

@@ -72,9 +72,6 @@ Always review and test generated code. This model is for **educational and resea
 ## How to Get Started with the Model
 
 ```python
-import re
-import time
-import sys
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 
@@ -88,17 +85,4 @@ model = PeftModel.from_pretrained(model, adapter_model)
 prompt = "Write a C program to calculate factorial of a number."
 inputs = tokenizer(prompt, return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=150)
-
-text = output[0]['generated_text']
-match = re.search(r'### Solution:\n```c\n(.*?)```', text, re.S)
-
-if match:
-    c_code = match.group(1)
-    print("Extracted C Code:\n")
-    for char in c_code:
-        sys.stdout.write(char)
-        sys.stdout.flush()
-        time.sleep(0.02)
-    print()
-else:
-    print("No C code found.")
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))