fix device yolo

cedula/app.py

@@ -109,7 +109,7 @@ def get_attributes(input_img):
     
     #images = check_image(input_img)
     images = [Image.fromarray(input_img)]
-    images = crop(weights="best.pt", files= images)
+    images = crop(weights="best.pt", files= images, device=device)
     #print(images)
     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
                                                               #entrega licencias

license/app.py

@@ -108,7 +108,7 @@ def get_attributes(input_img):
     model.to(device)
     
     images = [Image.fromarray(input_img)]
-    images = crop(weights="best.pt", files= images)
+    images = crop(weights="best.pt", files= images, device = device)
     image_cedula = [img[0] for img in images if img[1]==1][0] #0 en 'img[1]==0' es el label de cedula, si se reemplaza por 1
                                                               #entrega licencias
     
@@ -155,7 +155,6 @@ def create_model():
                         )
     return demo
 
-["examples/licencia.jpg"]
 if __name__ == '__main__':
     demo = create_model()
     demo.launch()