Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2025-01-07 10:12:58 +01:00 committed by GitHub
parent 96cd3895ff
commit e35cd0b490
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 5 additions and 9 deletions

View file

@ -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()