AnkitShrestha commited on
Commit
b0c7f29
·
1 Parent(s): dd4a9e3

Format reponse

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -201,7 +201,7 @@ async def batch_extract_text_vllm(background_tasks: BackgroundTasks, files: List
201
  outputs = llm.generate(inputs, sampling_params)
202
  for o in outputs:
203
  generated_text = o.outputs[0].text
204
- results.append(generated_text)
205
 
206
  logger.info(f"vLLM Batch processing completed in {time.time() - start_time:.2f} seconds")
207
  return {"extracted_texts": results}
 
201
  outputs = llm.generate(inputs, sampling_params)
202
  for o in outputs:
203
  generated_text = o.outputs[0].text
204
+ results.append(" ocr\n"+generated_text)
205
 
206
  logger.info(f"vLLM Batch processing completed in {time.time() - start_time:.2f} seconds")
207
  return {"extracted_texts": results}