Spaces:
Build error
Build error
diegulio
commited on
Commit
·
32d2c97
1
Parent(s):
cb46c78
Update app.py
Browse files- cedula/app.py +1 -1
cedula/app.py
CHANGED
@@ -106,7 +106,7 @@ def get_attributes(input_img):
|
|
106 |
images = [Image.fromarray(input_img)]
|
107 |
images = crop(weights="best.pt", files= images)
|
108 |
print(images)
|
109 |
-
image_cedula = [img[0] for img in images if img[1]==
|
110 |
#entrega licencias
|
111 |
|
112 |
pixel_values = processor(image_cedula, return_tensors="pt").pixel_values
|
|
|
106 |
images = [Image.fromarray(input_img)]
|
107 |
images = crop(weights="best.pt", files= images)
|
108 |
print(images)
|
109 |
+
image_cedula = [img[0] for img in images if img[1]==0][0] #0 en 'img[1]==0' es el label de cedula, si se reemplaza por 1
|
110 |
#entrega licencias
|
111 |
|
112 |
pixel_values = processor(image_cedula, return_tensors="pt").pixel_values
|