Damanger commited on
Commit
24927f2
·
1 Parent(s): c8282cf

corrigiendo app

Browse files
Files changed (2) hide show
  1. Dockerfile +3 -1
  2. app.py +16 -5
Dockerfile CHANGED
@@ -2,7 +2,9 @@ FROM python:3.11-slim
2
 
3
  ENV PYTHONUNBUFFERED=1 \
4
  HF_HOME=/tmp/.cache/huggingface \
5
- YOLO_CONFIG_DIR=/tmp/Ultralytics
 
 
6
 
7
  RUN apt-get update && apt-get install -y --no-install-recommends \
8
  libgl1 libglib2.0-0 && \
 
2
 
3
  ENV PYTHONUNBUFFERED=1 \
4
  HF_HOME=/tmp/.cache/huggingface \
5
+ YOLO_CONFIG_DIR=/tmp/Ultralytics \
6
+ MPLCONFIGDIR=/tmp/mplconfig \
7
+ HOME=/tmp
8
 
9
  RUN apt-get update && apt-get install -y --no-install-recommends \
10
  libgl1 libglib2.0-0 && \
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import os, io, base64, urllib.request, ssl, time, json
2
  from typing import Optional, List
3
  import numpy as np, cv2, torch
4
  from ultralytics import YOLO
@@ -9,8 +9,14 @@ from fastapi.middleware.cors import CORSMiddleware
9
  from pydantic import BaseModel, Field
10
 
11
  TMP_DIR = "/tmp"
12
- os.makedirs(f"{TMP_DIR}/Ultralytics", exist_ok=True)
13
- os.makedirs(f"{TMP_DIR}/.EasyOCR", exist_ok=True)
 
 
 
 
 
 
14
 
15
  from huggingface_hub import hf_hub_download
16
 
@@ -33,8 +39,13 @@ except Exception:
33
  yolo = YOLO(WEIGHTS)
34
 
35
  # EasyOCR con GPU si está disponible
36
- reader = easyocr.Reader(['en'], gpu=torch.cuda.is_available(),
37
- model_storage_directory=f"{TMP_DIR}/.EasyOCR")
 
 
 
 
 
38
 
39
  ALLOW = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
40
 
 
1
+ import os, io, base64, urllib.request, ssl, time, json, pathlib
2
  from typing import Optional, List
3
  import numpy as np, cv2, torch
4
  from ultralytics import YOLO
 
9
  from pydantic import BaseModel, Field
10
 
11
  TMP_DIR = "/tmp"
12
+ paths = [
13
+ f"{TMP_DIR}/Ultralytics",
14
+ f"{TMP_DIR}/.EasyOCR",
15
+ f"{TMP_DIR}/.EasyOCR/user_network",
16
+ f"{TMP_DIR}/mplconfig",
17
+ ]
18
+ for p in paths:
19
+ os.makedirs(p, exist_ok=True)
20
 
21
  from huggingface_hub import hf_hub_download
22
 
 
39
  yolo = YOLO(WEIGHTS)
40
 
41
  # EasyOCR con GPU si está disponible
42
+ reader = easyocr.Reader(
43
+ ['en'],
44
+ gpu=torch.cuda.is_available(),
45
+ model_storage_directory=f"{TMP_DIR}/.EasyOCR",
46
+ user_network_directory=f"{TMP_DIR}/.EasyOCR/user_network",
47
+ download_enabled=True,
48
+ )
49
 
50
  ALLOW = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
51