Spaces:
Sleeping
Sleeping
JoshuaCarrera commited on
Commit ·
773fc20
1
Parent(s): 4533622
Cambiar tamaño de las fotos
Browse files- 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=
|
| 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=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)
|