tamatwi commited on
Commit
094cf21
·
verified ·
1 Parent(s): e37fd6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import gradio as gr
5
 
6
  # PersonaHubデータセットの読み込み
7
- dataset = load_dataset("proj-persona/PersonaHub")
8
 
9
  # 日本語モデルを指定
10
  model_name = "rinna/japanese-gpt2-medium"
 
4
  import gradio as gr
5
 
6
  # PersonaHubデータセットの読み込み
7
+ dataset = load_dataset("proj-persona/PersonaHub", "math")
8
 
9
  # 日本語モデルを指定
10
  model_name = "rinna/japanese-gpt2-medium"