Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer
-import spaces
 @spaces.GPU
-# generator = pipeline('text-generation', model='distilgpt2', device=-1)
-# model_name = "rinna/japanese-gpt2-medium"
-tokenizer = AutoTokenizer.from_pretrained('rinna/japanese-gpt2-medium')
-generator = pipeline('text-generation', 'rinna/japanese-gpt2-medium', tokenizer=tokenizer, device=-1)
+# 日本語モデルを指定
+model_name = "rinna/japanese-gpt2-medium"
+
+# トークナイザーとパイプラインの設定
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=-1)
 
 def generate_text(prompt, max_length):
     result = generator(prompt, max_length=max_length, num_return_sequences=1)
@@ -14,8 +15,8 @@ def generate_text(prompt, max_length):
 iface = gr.Interface(
     fn=generate_text,
     inputs=[
-        gr.Textbox(label="プロンプト"),
-        gr.Slider(minimum=10, maximum=100, step=1, label="最大長")
+        gr.Textbox(label="プロンプト", placeholder="ここに日本語のプロンプトを入力してください"),
+        gr.Slider(minimum=10, maximum=200, value=50, step=1, label="最大長")
     ],
     outputs=gr.Textbox(label="生成されたテキスト")
 )