diff --git a/python/llm/example/CPU/PyTorch-Models/Model/llava/generate.py b/python/llm/example/CPU/PyTorch-Models/Model/llava/generate.py
index 1ce930dee7c..b2f81c36346 100644
--- a/python/llm/example/CPU/PyTorch-Models/Model/llava/generate.py
+++ b/python/llm/example/CPU/PyTorch-Models/Model/llava/generate.py
@@ -272,6 +272,9 @@ def get_stopping_criteria(conv, tokenizer, input_ids):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(
+
+
+if __name__ == '__main__':
         description='Predict Tokens using `generate()` API for LLaVA model')
     parser.add_argument('--repo-id-or-model-path', type=str, default="liuhaotian/llava-v1.5-13b",
                         help='The huggingface repo id for the LLaVA model to be downloaded'