Ultralytics Refactor https://ultralytics.com/actions (#18555)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
96cd3895ff
commit
e35cd0b490
5 changed files with 5 additions and 9 deletions
|
|
@ -184,12 +184,8 @@ class Inference:
|
|||
if __name__ == "__main__":
|
||||
import sys # Import the sys module for accessing command-line arguments
|
||||
|
||||
model = None # Initialize the model variable as None
|
||||
|
||||
# Check if a model name is provided as a command-line argument
|
||||
args = len(sys.argv)
|
||||
if args > 1:
|
||||
model = sys.argv[1] # Assign the first argument as the model name
|
||||
|
||||
model = sys.argv[1] if args > 1 else None # assign first argument as the model name
|
||||
# Create an instance of the Inference class and run inference
|
||||
Inference(model=model).inference()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue