Spaces:
Sleeping
Sleeping
JoshuaCarrera commited on
Commit 路
c3c8af8
1
Parent(s): 98c4eab
Agilizar respuestas
Browse files- Dockerfile +1 -1
- src/app/app.py +2 -2
Dockerfile
CHANGED
|
@@ -37,4 +37,4 @@ COPY --chown=user . .
|
|
| 37 |
EXPOSE 7860
|
| 38 |
|
| 39 |
# 7. Arranque optimizado para 16GB de RAM (4 workers)
|
| 40 |
-
CMD ["uvicorn", "src.app.app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "
|
|
|
|
| 37 |
EXPOSE 7860
|
| 38 |
|
| 39 |
# 7. Arranque optimizado para 16GB de RAM (4 workers)
|
| 40 |
+
CMD ["uvicorn", "src.app.app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
|
src/app/app.py
CHANGED
|
@@ -130,7 +130,7 @@ def detection_vef(image: UploadFile):
|
|
| 130 |
debug_buffer = io.BytesIO()
|
| 131 |
imageFile.save(debug_buffer, format="JPEG")
|
| 132 |
|
| 133 |
-
which_currency = models["INFERENCIA"].predict(imageFile, verbose=False, imgsz=
|
| 134 |
|
| 135 |
if len(which_currency[0].boxes) == 0:
|
| 136 |
return {"message": "No objects detected"}
|
|
@@ -142,7 +142,7 @@ def detection_vef(image: UploadFile):
|
|
| 142 |
else:
|
| 143 |
currency = "USD"
|
| 144 |
|
| 145 |
-
results = models[currency].predict(imageFile, verbose=False, imgsz=
|
| 146 |
|
| 147 |
if len(results[0].boxes) > 0:
|
| 148 |
# Tomamos el label y la confianza de la primera detecci贸n (la de mayor confianza usualmente)
|
|
|
|
| 130 |
debug_buffer = io.BytesIO()
|
| 131 |
imageFile.save(debug_buffer, format="JPEG")
|
| 132 |
|
| 133 |
+
which_currency = models["INFERENCIA"].predict(imageFile, verbose=False, imgsz=320, conf=0.25)
|
| 134 |
|
| 135 |
if len(which_currency[0].boxes) == 0:
|
| 136 |
return {"message": "No objects detected"}
|
|
|
|
| 142 |
else:
|
| 143 |
currency = "USD"
|
| 144 |
|
| 145 |
+
results = models[currency].predict(imageFile, verbose=False, imgsz=320, conf=0.25)
|
| 146 |
|
| 147 |
if len(results[0].boxes) > 0:
|
| 148 |
# Tomamos el label y la confianza de la primera detecci贸n (la de mayor confianza usualmente)
|