tamatwi commited on
Commit
94942ab
·
verified ·
1 Parent(s): 695e718

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -4,9 +4,6 @@ from transformers import pipeline, AutoTokenizer
4
  import gradio as gr
5
 
6
  @GPU(duration=120)
7
- def generate_text(prompt, max_length):
8
- result = generator(prompt, max_length=max_length, num_return_sequences=1)
9
- return result[0]['generated_text']
10
 
11
  # 日本語モデルを指定
12
  model_name = "rinna/japanese-gpt2-medium"
 
4
  import gradio as gr
5
 
6
  @GPU(duration=120)
 
 
 
7
 
8
  # 日本語モデルを指定
9
  model_name = "rinna/japanese-gpt2-medium"