Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"
|