alexfremont commited on
Commit
0754334
·
1 Parent(s): 7f89f2f

test new fix

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -195,7 +195,9 @@ async def batch_predict(request: BatchPredictRequest):
195
  with ProcessPoolExecutor() as executor:
196
  results = list(
197
  executor.map(
198
- process_single_image, [(url, model) for url in request.imageUrls]
 
 
199
  )
200
  )
201
 
 
195
  with ProcessPoolExecutor() as executor:
196
  results = list(
197
  executor.map(
198
+ process_single_image,
199
+ request.imageUrls,
200
+ [model] * len(request.imageUrls),
201
  )
202
  )
203