Spaces:
Runtime error
Runtime error
cannot pickle 'generator' object
Browse files
app.py
CHANGED
@@ -12,7 +12,9 @@ import spaces
|
|
12 |
def run_on_gpu(input_shape, text, seed, guidance_scale):
|
13 |
from model import Model
|
14 |
model = Model()
|
15 |
-
|
|
|
|
|
16 |
|
17 |
|
18 |
def main():
|
|
|
12 |
def run_on_gpu(input_shape, text, seed, guidance_scale):
|
13 |
from model import Model
|
14 |
model = Model()
|
15 |
+
res_generator = model.run(shape_path=input_shape, text=text, seed=seed, guidance_scale=guidance_scale)
|
16 |
+
return list(res_generator)
|
17 |
+
|
18 |
|
19 |
|
20 |
def main():
|