JoshuaCarrera commited on
Commit
773fc20
·
1 Parent(s): 4533622

Cambiar tamaño de las fotos

Browse files
Files changed (1) hide show
  1. src/app/app.py +2 -2
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=640, 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=640, 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=416, 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=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)