nyasukun commited on
Commit
2ee9d24
·
verified ·
1 Parent(s): 969d05b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)