Commit
·
e76352d
1
Parent(s):
0df3f4e
set number of workers
Browse files
app.py
CHANGED
|
@@ -74,7 +74,7 @@ def _predict(inputs):
|
|
| 74 |
predicted_images = []
|
| 75 |
images = list(pluck(1, images_urls))
|
| 76 |
urls = list(pluck(0, images_urls))
|
| 77 |
-
predictions = classif_pipeline(images, top_k=1)
|
| 78 |
for url, pred in zip(urls, predictions):
|
| 79 |
top_pred = pred[0]
|
| 80 |
if top_pred['label'] == 'illustrated':
|
|
|
|
| 74 |
predicted_images = []
|
| 75 |
images = list(pluck(1, images_urls))
|
| 76 |
urls = list(pluck(0, images_urls))
|
| 77 |
+
predictions = classif_pipeline(images, top_k=1, num_workers=2)
|
| 78 |
for url, pred in zip(urls, predictions):
|
| 79 |
top_pred = pred[0]
|
| 80 |
if top_pred['label'] == 'illustrated':
|