Update README.md

README.md

@@ -49,16 +49,22 @@ Due to the diverse nature of the training data, which includes websites, books,
 import torch
 from transformers import AutoTokenizer, GPT2LMHeadModel
 from transformers import pipeline
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 device_id = 0 if torch.cuda.is_available() else -1
+
 model = GPT2LMHeadModel.from_pretrained("ytu-ce-cosmos/turkish-gpt2-medium-350m-instruct-v0.1").to(device)
+
 tokenizer = AutoTokenizer.from_pretrained("ytu-ce-cosmos/turkish-gpt2-medium-350m-instruct-v0.1")
+
 text_generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device=device_id, max_new_tokens=256)
+
 def get_model_response(instruction):
     instruction_prompt = f"### Kullanıcı:\n{instruction}\n### Asistan:\n"
     result = text_generator(instruction_prompt)
     generated_response = result[0]['generated_text']
     return generated_response[len(instruction_prompt):]
+
 model_response = get_model_response("Evde egzersiz yapmanın avantajlarını açıkla.")
 print(model_response)
 """
@@ -70,9 +76,11 @@ To use the chat template:
 
 ```python
 chat_generator = pipeline("conversational", model=model, tokenizer=tokenizer, device=device_id, max_new_tokens=256)
+
 messages = [
     {"role": "user", "content": "Evde egzersiz yapmanın avantajlarını açıkla."}
 ]
+
 chat_outputs = chat_generator(messages)
 print(chat_outputs)
 """