alonb19 commited on
Commit
8bef43b
·
verified ·
1 Parent(s): fbfec87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -1,16 +1,21 @@
1
  from fastapi import FastAPI, UploadFile, File, HTTPException
2
  from fastapi.responses import JSONResponse
3
- import easyocr
4
  import numpy as np
5
  from PIL import Image
6
  import io
7
  import uvicorn
8
  import os
9
 
 
 
 
 
 
 
10
  app = FastAPI()
11
 
12
  # Inicializar el lector OCR
13
- reader = easyocr.Reader(['es', 'en'], model_storage_directory='/app/model')
14
 
15
  @app.post("/ocr")
16
  async def extract_text(file: UploadFile = File(...)):
 
1
  from fastapi import FastAPI, UploadFile, File, HTTPException
2
  from fastapi.responses import JSONResponse
 
3
  import numpy as np
4
  from PIL import Image
5
  import io
6
  import uvicorn
7
  import os
8
 
9
+ # Configurar variables de entorno para EasyOCR antes de importarlo
10
+ os.environ['EASYOCR_MODULE_PATH'] = '/app/model'
11
+ os.environ['HOME'] = '/app'
12
+
13
+ import easyocr
14
+
15
  app = FastAPI()
16
 
17
  # Inicializar el lector OCR
18
+ reader = easyocr.Reader(['es', 'en'])
19
 
20
  @app.post("/ocr")
21
  async def extract_text(file: UploadFile = File(...)):