multimodalart HF staff commited on
Commit
b4ac571
1 Parent(s): 3416183

suggested patching

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -16,6 +16,8 @@ omni_zero = OmniZeroCouple(
16
  device="cuda",
17
  )
18
 
 
 
19
  @spaces.GPU()
20
  def generate(
21
  base_image="https://cdn-prod.styleof.com/inferences/cm1ho5cjl14nh14jec6phg2h8/i6k59e7gpsr45ufc7l8kun0g-medium.jpeg",
 
16
  device="cuda",
17
  )
18
 
19
+ omni_zero.generate = spaces.GPU(omni_zero.generate)
20
+
21
  @spaces.GPU()
22
  def generate(
23
  base_image="https://cdn-prod.styleof.com/inferences/cm1ho5cjl14nh14jec6phg2h8/i6k59e7gpsr45ufc7l8kun0g-medium.jpeg",