Yaron Koresh commited on
Commit
2202700
·
verified ·
1 Parent(s): c728865

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -191,7 +191,7 @@ def run(p1,p2,*result):
191
  rng = list(range(ln))
192
  arr1 = [p for _ in rng]
193
  print("Starting pool!")
194
- pool = Pool()
195
  out1 = list(pool.imap(infer1,arr1))
196
  arr2 = [{"a":p1_en,"b":p2_en,"c":out1[_]} for _ in rng]
197
  out2 = list(pool.imap(infer2,arr2))
 
191
  rng = list(range(ln))
192
  arr1 = [p for _ in rng]
193
  print("Starting pool!")
194
+ pool = Pool(nodes=ln*2)
195
  out1 = list(pool.imap(infer1,arr1))
196
  arr2 = [{"a":p1_en,"b":p2_en,"c":out1[_]} for _ in rng]
197
  out2 = list(pool.imap(infer2,arr2))