Spaces:
Runtime error
Runtime error
Damanger
commited on
Commit
·
24927f2
1
Parent(s):
c8282cf
corrigiendo app
Browse files- Dockerfile +3 -1
- 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 |
-
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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(
|
| 37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|