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
Files changed (1) hide show
  1. app.py +2 -1
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="cuda:0")
 
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):