Sergey Mikhno
commited on
Commit
·
5576d76
1
Parent(s):
bd0c3e4
try cuda
Browse files- app/app.py +1 -1
app/app.py
CHANGED
|
@@ -39,7 +39,7 @@ def doc_layout():
|
|
| 39 |
filename, # Image to predict
|
| 40 |
imgsz=1024, # Prediction image size
|
| 41 |
conf=0.2, # Confidence threshold
|
| 42 |
-
device="
|
| 43 |
)
|
| 44 |
names = det_res[0].names
|
| 45 |
blocknames = [names[int(n)] for n in det_res[0].boxes.cls]
|
|
|
|
| 39 |
filename, # Image to predict
|
| 40 |
imgsz=1024, # Prediction image size
|
| 41 |
conf=0.2, # Confidence threshold
|
| 42 |
+
device="cuda:0" # Device to use (e.g., 'cuda:0' or 'cpu')
|
| 43 |
)
|
| 44 |
names = det_res[0].names
|
| 45 |
blocknames = [names[int(n)] for n in det_res[0].boxes.cls]
|