zero gpu

app.py

@@ -1,14 +1,17 @@
 import gradio as gr
 from PIL import Image
 import os
+import spaces
 
-os.environ['CUDA_VISIBLE_DEVICES'] = '7'
+# os.environ['CUDA_VISIBLE_DEVICES'] = '7'
 
 from OmniGen import OmniGenPipeline
 
 pipe = OmniGenPipeline.from_pretrained("shitao/tmp-preview")
+pipe.to("cuda")
 
 # 示例处理函数：生成图像
+@spaces.GPU
 def generate_image(text, img1, img2, img3, height, width, guidance_scale):
     input_images = [img1, img2, img3]
     # 去除 None