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