Spaces:

DarianT
/

ID-Booth

Running on Zero

DarianT commited on May 21

Commit

638c39d

1 Parent(s): 15b4f9a

Update to use spaces.GPU

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 import numpy as np
 import random
-# import spaces  # Uncomment if using ZeroGPU
 from diffusers import DiffusionPipeline
 import torch
@@ -22,7 +22,7 @@ gender_dict = {"ID_0": "male"}
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 512
-# @spaces.GPU  # Uncomment if using ZeroGPU
 def infer(
     background,
     pose,

 import gradio as gr
 import numpy as np
 import random
+import spaces  # Uncomment if using ZeroGPU
 from diffusers import DiffusionPipeline
 import torch
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 512
+@spaces.GPU  # Uncomment if using ZeroGPU
 def infer(
     background,
     pose,