Update README.md

README.md

@@ -41,14 +41,13 @@ def generate_response(model, tokenizer, prompt, device):
         device: テンソルを転送するデバイス（例: "cuda" や "cpu"）
 
     Returns:
-        生成されたテキスト応答（特殊トークンを除外してデコード済みの文字列）
+        生成されたテキスト応答（デコード済みの文字列）
     """
     inputs = tokenizer(prompt, return_tensors="pt") # テンソル変換
     inputs = {key: tensor.to(device) for key, tensor in inputs.items()} # デバイスに転送
     output = model.generate(**inputs, streamer=streamer) # TextStreamerを用いてテキスト生成を実行
     return tokenizer.decode(output[0], skip_special_tokens=True, max_new_tokens=1024) # 生成されたトークンをデコード
 
-
 device = "cuda" # 使用するデバイスを "cuda" （GPU）に設定
 prompt = "AIについて簡潔に教えてください。" # テキスト生成用の入力プロンプトを定義