diff --git a/cogrun.py b/cogrun.py index e437b82..0068bf1 100644 --- a/cogrun.py +++ b/cogrun.py @@ -27,7 +27,7 @@ class Predictor(BasePredictor): default=-1 ), intermediate_image_count: int = Input( - description='Set the number of intermediate images to show', + description='Set the number of intermediate images to show while running', choices=[1, 2, 4, 8, 16], default=8 ),