Update model.py

Ziad-Algrafi · May 19, 2024 · 31557fc · 31557fc
1 parent 96fdc44
commit 31557fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yolo_world_onnx/model.py b/yolo_world_onnx/model.py
@@ -22,7 +22,7 @@ def select_device(device):
     elif device == "cpu":
         return "cpu"
     else:
-        print("Invalid device input or GPU is not Available. For chossing a CPU device enter 'cpu' and for a GPU enters '0', defaulting to a cpu")
+        print("Invalid device input or GPU is not Available. For choosing a CPU device enter 'cpu' and for a GPU enters '0', defaulting to a cpu")
         return "cpu"
 
 class TextEmbedder:
@@ -203,4 +203,4 @@ def get_input_details(self):
 
     def get_output_details(self):
         model_outputs = self.session.get_outputs()
-        self.output_names = [model_outputs[i].name for i in range(len(model_outputs))]
+        self.output_names = [model_outputs[i].name for i in range(len(model_outputs))]