Update app.py

app.py

@@ -2,12 +2,12 @@ import gradio
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModelForCausalLM.from_pretrained("cyberagent/open-calm-small", device_map="auto", torch_dtype=torch.float32)
-tokenizer = AutoTokenizer.from_pretrained("cyberagent/open-calm-small")
+model = AutoModelForCausalLM.from_pretrained("cyberagent/open-calm-large", device_map="auto", torch_dtype=torch.float32)
+tokenizer = AutoTokenizer.from_pretrained("cyberagent/open-calm-large")
 
 def my_inference_function(mail):
     mail=mail.replace("\n", "")
-    inputs = tokenizer(f"(上司へのメール敬語で){mail}", return_tensors="pt").to(model.device)
+    inputs = tokenizer(mail, return_tensors="pt").to(model.device)
     with torch.no_grad():
         tokens = model.generate(
             **inputs,