Update app.py

app.py

@@ -1,6 +1,6 @@
 import time
 import torch
-# import spaces
+import spaces
 from PIL import Image
 from tqdm import tqdm
 from threading import Thread
@@ -71,7 +71,7 @@ vqllm = vqllm.eval()
 
 num_codebooks = vae_cfg.num_codebooks
 
-# @spaces.GPU
+@spaces.GPU
 def bot_streaming_I2T(message, history):
     print(message)
     global stop_flag
@@ -149,7 +149,7 @@ def show_gallery(images):
     gallery = gr.Gallery(images, label="Gallery", columns=4, height="auto",preview=True,scale=0.05)  # 设置两行两列的布局
     return gallery
 
-# @spaces.GPU
+@spaces.GPU
 def bot_streaming_T2I(message, history,guidance_scale, temperature, top_K, top_P):
     global stop_flag
     stop_flag = True
@@ -269,7 +269,6 @@ with gr.Blocks(fill_height=True) as demo:
         aaa = gr.ChatInterface(
         fn=bot_streaming_T2I,
         examples=[
-            ["Magic mushroom in the forest", 5.0, 1.0, 2048, 1.0],
             ["cherry tree on the surface of the moon", 5.0, 1.0, 2048, 1.0],
             ["New York City at night with starry night vincent van gogh style", 5.0, 1.0, 2048, 1.0],
             ["cavalier king charles spaniel being cute and ultra realistic with cute sunglasses", 5.0, 1.0, 2048, 1.0],