JoshuaCarrera commited on
Commit
c3c8af8
路
1 Parent(s): 98c4eab

Agilizar respuestas

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. 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", "2"]
 
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=416, conf=0.25)
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=416, 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)
 
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)