diff --git a/run_model.py b/run_model.py index 0f723e4..5f1a8d0 100644 --- a/run_model.py +++ b/run_model.py @@ -1,7 +1,7 @@ from tinyllava.eval.run_tiny_llava import * import argparse - +parser = argparse.ArgumentParser() parser.add_argument('--model_path', type=str, required=True, help="Path to the model.") parser.add_argument('--src', type=str, required=True, help="Path of Input Pictures and Reference Codes.") parser.add_argument('--out', type=str, required=True, help="Any output path you like.") @@ -187,4 +187,4 @@ for index in range(len(input_types)): print() print("Can't Generate these inputs:") -print(errors) \ No newline at end of file +print(errors)