haohsiang commited on
Commit
c77a568
·
1 Parent(s): 54e000b

add ZeroGPU decoration

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  import gradio as gr
3
  import torch
@@ -122,6 +123,7 @@ def translate_to_english(text):
122
  generated_images = None
123
  last_genimg_times = 0
124
 
 
125
  def generate_images(history):
126
  global generated_images # 使用全域變數來儲存圖片
127
  global last_genimg_times
 
1
+ import spaces
2
  import os
3
  import gradio as gr
4
  import torch
 
123
  generated_images = None
124
  last_genimg_times = 0
125
 
126
+ @spaces.GPU(duration=120)
127
  def generate_images(history):
128
  global generated_images # 使用全域變數來儲存圖片
129
  global last_genimg_times