Spaces:
Build error
Build error
diegulio
commited on
Commit
·
aa401bb
1
Parent(s):
31f55ae
print device
Browse files- cedula/app.py +1 -0
- license/app.py +1 -0
cedula/app.py
CHANGED
|
@@ -106,6 +106,7 @@ def get_attributes(input_img):
|
|
| 106 |
|
| 107 |
model.eval()
|
| 108 |
model.to(device)
|
|
|
|
| 109 |
|
| 110 |
#images = check_image(input_img)
|
| 111 |
images = [Image.fromarray(input_img)]
|
|
|
|
| 106 |
|
| 107 |
model.eval()
|
| 108 |
model.to(device)
|
| 109 |
+
print(device)
|
| 110 |
|
| 111 |
#images = check_image(input_img)
|
| 112 |
images = [Image.fromarray(input_img)]
|
license/app.py
CHANGED
|
@@ -106,6 +106,7 @@ def get_attributes(input_img):
|
|
| 106 |
|
| 107 |
model.eval()
|
| 108 |
model.to(device)
|
|
|
|
| 109 |
|
| 110 |
images = [Image.fromarray(input_img)]
|
| 111 |
images = crop(weights="best.pt", files= images, device = device)
|
|
|
|
| 106 |
|
| 107 |
model.eval()
|
| 108 |
model.to(device)
|
| 109 |
+
print(device)
|
| 110 |
|
| 111 |
images = [Image.fromarray(input_img)]
|
| 112 |
images = crop(weights="best.pt", files= images, device = device)
|