Spaces:
Sleeping
Sleeping
TroglodyteDerivations
commited on
Commit
•
bdc8c91
1
Parent(s):
7109c38
Updated line 21 with: self.model = yolov9.load(self.model_path, device="cpu") # Inference generated from CPU (And a # on line 22)
Browse files
app.py
CHANGED
@@ -18,7 +18,8 @@ class Inference_Nascent_Spawning_Deriving_From_YOLOv9:
|
|
18 |
# Object behavior / Method -> 2
|
19 |
def load_model(self, model_id):
|
20 |
self.model_path = self.download_models(model_id)
|
21 |
-
self.model = yolov9.load(self.model_path, device="
|
|
|
22 |
|
23 |
# Object behavior / Method -> 3
|
24 |
def configure_model(self, conf_threshold, iou_threshold):
|
|
|
18 |
# Object behavior / Method -> 2
|
19 |
def load_model(self, model_id):
|
20 |
self.model_path = self.download_models(model_id)
|
21 |
+
self.model = yolov9.load(self.model_path, device="cpu") # Inference generated from CPU
|
22 |
+
#self.model = yolov9.load(self.model_path, device="cuda:0")
|
23 |
|
24 |
# Object behavior / Method -> 3
|
25 |
def configure_model(self, conf_threshold, iou_threshold):
|