Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
b63699d
1
Parent(s):
7bed26b
Updated requirements
Browse files- app.py +1 -0
- requirements.txt +3 -0
app.py
CHANGED
|
@@ -13,6 +13,7 @@ def predict_image(img, conf_threshold, iou_threshold):
|
|
| 13 |
|
| 14 |
results = model.predict(
|
| 15 |
source=img,
|
|
|
|
| 16 |
conf=conf_threshold,
|
| 17 |
iou=iou_threshold,
|
| 18 |
show_labels=True,
|
|
|
|
| 13 |
|
| 14 |
results = model.predict(
|
| 15 |
source=img,
|
| 16 |
+
device='cuda:0',
|
| 17 |
conf=conf_threshold,
|
| 18 |
iou=iou_threshold,
|
| 19 |
show_labels=True,
|
requirements.txt
CHANGED
|
@@ -1 +1,4 @@
|
|
| 1 |
ultralytics
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
ultralytics
|
| 2 |
+
torch
|
| 3 |
+
supervision
|
| 4 |
+
spaces
|