Update app.py

app.py

@@ -6,7 +6,7 @@ model_name = "cameltech/japanese-gpt-1b-PII-masking"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-instruction = "文中の個人情報をマスキングせよ\n\n"
+instruction = "# タスク\n入力文中の個人情報をマスキングせよ\n\n# 入力文\n"
 
 if torch.cuda.is_available():
     model = model.to("cuda")
@@ -28,6 +28,7 @@ generation_config = {
 }
 
 def generate(input_text):
+    input_text = instruction + input_text
     input_text += tokenizer.eos_token
     input_text = preprocess(input_text)