Update script.py

script.py

@@ -18,22 +18,29 @@ from llava.constants import (
 )
 from llava.conversation import conv_templates, SeparatorStyle
 
+import argparse
 import torch
 import requests
 from PIL import Image
 from io import BytesIO
 import re
 
+parser = argparse.ArgumentParser()
+parser.add_argument("--model-path", type=str, default="liuhaotian/llava-v1.6-mistral-7b")
+parser.add_argument("--image-file", type=str, required=True)
+parser.add_argument("--inference-type", type=str, default="auto")
+parser.add_argument("--prompt", type=str, default="Explain this image")
+cmd_args = parser.parse_args()
 
 # Line 138 uncomment the cuda() to use GPUs
 
 # device = "cpu"
-device = "auto"
+device = cmd_args.inference_type
 
-prompt = "What are the things I should be cautious about when I visit here?"
-image_file = "Great-Room-4.jpg"
+prompt = cmd_args.prompt
+image_file = cmd_args.image_file
 
-model_path = "liuhaotian/llava-v1.6-mistral-7b" # <path to model weights>
+model_path = cmd_args.model_path