Update infer.py

infer.py

@@ -10,7 +10,7 @@ from PIL import Image, ImageDraw
 parser = argparse.ArgumentParser(description="Qwen-VL Inference")
 parser.add_argument("--image_path", type=str, required=False, default="./test_screenshots/amazon.png", help="Path to the input image")
 parser.add_argument("--query", type=str, required=False, default="Click on the 'Chairs'.", help="Text query or instruction")
-parser.add_argument("--model_dir", type=str, default="./showui-2b-awq-4bit", help="Path to the local ShowUI model directory")
+parser.add_argument("--model_dir", type=str, default="./", help="Path to the local ShowUI model directory")
 args = parser.parse_args()
 
 DEVICE = "cuda:0"