Hugging Face's logo

Spaces:
ImageIN
/
iiif_book_manifest_illustration_detection
Runtime error

App Files Community
davanstrien HF Staff commited on
Commit
e76352d
·
1 Parent(s): 0df3f4e

set number of workers

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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':