pizzagatakasugi commited on
Commit
26ed1ea
·
1 Parent(s): 337557b

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +1 -1
app2.py CHANGED
@@ -6,7 +6,7 @@ import pandas as pd
6
 
7
  #モデルの読み込み
8
  tokenizer = T5Tokenizer.from_pretrained("pizzagatakasugi/shogi_t5", is_fast=True)
9
- model = T5ForConditionalGeneration.from_pretrained("pizzagatakasugi/shogi_t5")
10
  model.eval()
11
 
12
  st.title("将棋解説文の自動生成")
 
6
 
7
  #モデルの読み込み
8
  tokenizer = T5Tokenizer.from_pretrained("pizzagatakasugi/shogi_t5", is_fast=True)
9
+ model = T5ForConditionalGeneration.from_pretrained("pizzagatakasugi/shogi_t5_v1")
10
  model.eval()
11
 
12
  st.title("将棋解説文の自動生成")