Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ generator = pipeline(
|
|
18 |
)
|
19 |
|
20 |
# GPUを利用する推論関数を定義(ZeroGPUのためデコレータを使用)
|
21 |
-
@spaces.GPU
|
22 |
def generate_text(prompt):
|
23 |
# プロンプトからテキストを生成し、結果文字列を返す
|
24 |
result = generator(prompt, max_new_tokens=100, do_sample=True)
|
|
|
18 |
)
|
19 |
|
20 |
# GPUを利用する推論関数を定義(ZeroGPUのためデコレータを使用)
|
21 |
+
@spaces.GPU(duration=120)
|
22 |
def generate_text(prompt):
|
23 |
# プロンプトからテキストを生成し、結果文字列を返す
|
24 |
result = generator(prompt, max_new_tokens=100, do_sample=True)
|