Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
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 |
-
|
196 |
-
|
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 |
-
|
203 |
-
|
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():
|