Yaron Koresh commited on
Commit
55dfaa6
·
verified ·
1 Parent(s): abc06e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -8
app.py CHANGED
@@ -192,17 +192,13 @@ def run(p1,p2,*result):
192
  arr1 = [p for _ in rng]
193
  out1 = None
194
  print("Starting pool!")
195
- pool1 = Pool(ln)
196
- out1 = pool1.map(infer1,arr1)
197
- pool1.close()
198
- pool1.join()
199
  lst = list(out1)
200
  arr2 = [{"a":p1_en,"b":p2_en,"c":lst[_]} for _ in rng]
201
  out2 = None
202
- pool2 = Pool(ln)
203
- out2 = pool2.map(infer2,arr2)
204
- pool2.close()
205
- pool2.join()
206
  return list(out2)
207
 
208
  def ui():
 
192
  arr1 = [p for _ in rng]
193
  out1 = None
194
  print("Starting pool!")
195
+ with Pool(ln) as pool1:
196
+ out1 = pool1.map(infer1,arr1)
 
 
197
  lst = list(out1)
198
  arr2 = [{"a":p1_en,"b":p2_en,"c":lst[_]} for _ in rng]
199
  out2 = None
200
+ with Pool(ln) as pool2:
201
+ out2 = pool2.map(infer2,arr2)
 
 
202
  return list(out2)
203
 
204
  def ui():