Spaces:

tamatwi
/

shortrinnashortgpu_rinna

App Files Files Community

tamatwi commited on Jul 6

Commit

69169eb

•

1 Parent(s): 04ed384

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 model_name = "rinna/japanese-gpt2-medium"
 # トークナイザーとパイプラインの設定
-tokenizer = AutoTokenizer.from_pretrained(model_name)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)

 model_name = "rinna/japanese-gpt2-medium"
 # トークナイザーとパイプラインの設定
+tokenizer = AutoTokenizer.from_pretrained(model_name, legacy=False)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)