Spaces:
Sleeping
Sleeping
Commit
·
b0c7f29
1
Parent(s):
dd4a9e3
Format reponse
Browse files
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}
|