PuLID

Runtime error

yanze commited on May 1

Commit

4eca20b

•

1 Parent(s): bb7359c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,3 +1,4 @@
 import gradio as gr
 import numpy as np
 import torch
@@ -19,6 +20,7 @@ DEFAULT_NEGATIVE_PROMPT = (
 )
 def run(*args):
     id_image = args[0]
     supp_images = args[1:4]

+import spaces
 import gradio as gr
 import numpy as np
 import torch
 )
+@spaces.GPU
 def run(*args):
     id_image = args[0]
     supp_images = args[1:4]