machuofan commited on
Commit
57437f3
·
verified ·
1 Parent(s): 3dc7bcd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import time
2
  import torch
3
- import spaces
4
  from PIL import Image
5
  from tqdm import tqdm
6
  from threading import Thread
@@ -71,7 +71,7 @@ vqllm = vqllm.eval()
71
 
72
  num_codebooks = vae_cfg.num_codebooks
73
 
74
- @spaces.GPU
75
  def bot_streaming_I2T(message, history):
76
  print(message)
77
  global stop_flag
@@ -149,7 +149,7 @@ def show_gallery(images):
149
  gallery = gr.Gallery(images, label="Gallery", columns=4, height="auto",preview=True,scale=0.05) # 设置两行两列的布局
150
  return gallery
151
 
152
- @spaces.GPU
153
  def bot_streaming_T2I(message, history,guidance_scale, temperature, top_K, top_P):
154
  global stop_flag
155
  stop_flag = True
 
1
  import time
2
  import torch
3
+ # import spaces
4
  from PIL import Image
5
  from tqdm import tqdm
6
  from threading import Thread
 
71
 
72
  num_codebooks = vae_cfg.num_codebooks
73
 
74
+ # @spaces.GPU
75
  def bot_streaming_I2T(message, history):
76
  print(message)
77
  global stop_flag
 
149
  gallery = gr.Gallery(images, label="Gallery", columns=4, height="auto",preview=True,scale=0.05) # 设置两行两列的布局
150
  return gallery
151
 
152
+ # @spaces.GPU
153
  def bot_streaming_T2I(message, history,guidance_scale, temperature, top_K, top_P):
154
  global stop_flag
155
  stop_flag = True