Spaces:

Yuanshi
/

OminiControl_Art

Running on Zero

Yuanshi commited on Apr 8

Commit

cec983f

verified ·

1 Parent(s): 629e679

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -156,7 +156,7 @@ def infer(
 if USE_ZERO_GPU:
-    infer = spaces.GPU(infer, duration=360)
 if __name__ == "__main__":
     demo = gradio_interface()

 if USE_ZERO_GPU:
+    infer = spaces.GPU(infer)
 if __name__ == "__main__":
     demo = gradio_interface()