perler commited on
Commit
b46e8b2
·
1 Parent(s): 18eb0a8

cannot pickle 'generator' object

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
- return model.run(shape_path=input_shape, text=text, seed=seed, guidance_scale=guidance_scale)
 
 
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():